DeepLearning/Pytorch (2) 썸네일형 리스트형 [Pytorch] Anaconda 가상환경 + Pytorch GPU 설치 대학원 일과를 끝나고 정리할 겸 집에서 이 글을 작성하게 되었다(졸리다...) Pytorch를 GPU 환경에서 돌리는 것은 은근 까다롭지만 Tensorflow만큼은 아니니, 어떻게든 따라올 수 있으면 좋겠다! - 컴퓨터 사양 : Window 11 + RTX 3070 - CUDA Version : 11.6 - CuDNN Version : 8.4.0 사전작업 - CUDA, CuDNN 설치 https://taichi1.tistory.com/3 [CUDA] RTX3070 + Window11 CUDA, CuDNN 설정 RTX 3070에서 Tensorflow와 Pytorch를 사용하기 위해 CUDA와 cuDNN 환경 세팅을 해보았다! CUDA 설치 https://developer.nvidia.com/cuda-too.. [pytorch] item() 함수 tensor에 하나의 값만 존재한다면, item() 함수를 사용해 python scalar 값을 얻을 수 있다. tensor에 여러개의 값이 존재한다면, 사용이 불가능하다. 아래 예시 코드를 참고해보자. import torch x = torch.randn(1) print(x) print(x.item()) 결과는 다음과 같은 것을 알 수가 있다. 이전 1 다음
