1. 입문과정(Mnist)

https://blog.nerdfactory.ai/2020/10/08/Tutorial-1-for-Deep-Learning-Model-Trainer-Development.html

 

너드팩토리

너드팩토리에서 운영하는 블로그 입니다.

blog.nerdfactory.ai

2. Criterion, Loss Function정리

[PyTorch] 자주쓰는 Loss Function (Cross-Entropy, MSE) 정리 - ZZEN’s Blog (nuguziii.github.io)

 

[PyTorch] 자주쓰는 Loss Function (Cross-Entropy, MSE) 정리

PyTorch에서 제가 개인적으로 자주쓰는 Loss Function (Cross Entropy, MSE) 들을 정리한 글입니다.

nuguziii.github.io

 

'PyTorch' 카테고리의 다른 글

설치된 파이토치 버전보기  (0) 2022.08.30
PyTorch 쿠다 환경 구성 (gpu 사용하기)  (0) 2022.08.28

 

 

import torch

print(torch.__version__)

'PyTorch' 카테고리의 다른 글

[Pytorch] 기본기  (0) 2023.06.11
PyTorch 쿠다 환경 구성 (gpu 사용하기)  (0) 2022.08.28

 

 

1. CUDA Toolkit down (필자 test는rtx 3060,  11.3.0버전) https://developer.nvidia.com/cuda-toolkit-archive 

 

CUDA Toolkit Archive

Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production

developer.nvidia.com

2. cuDNN 설치(필자는 11.3버전) https://developer.nvidia.com/cudnn

3. 다운 받은 cuDNN압축풀어서 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3 경로에 넣어주기

4. 파이토치설치: pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

* 1번 CUDA설치가 되지 않으면 NVIDIA그래픽드라이버를 지우고 다시 설치

*  1번 CUDA를 설치하려는데 다른 설치프로그램이 있다고 하면 서비스에서 windows installer를 정지했다 재시작 후 재부팅하고 실행

5. cmd에서 파이썬 실행하고

>>> import torch

>>> torch.cuda.get_device_name(0)

했을 때 해당 그래픽카드가 검색되면 설치 완료

 

아래 참고 블로그에 자세한 설명 있음

참고: https://leonam.tistory.com/98

'PyTorch' 카테고리의 다른 글

[Pytorch] 기본기  (0) 2023.06.11
설치된 파이토치 버전보기  (0) 2022.08.30

+ Recent posts