TORCH
torch version = 1.10.0에서 GPU 설정하기
난 pip로 깔았으나 conda를 사용하는 사람은 conda 누르고 깔면 됨.
아래를 terminal에 깔아준다.
'
pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f
https://download.pytorch.org/whl/cu102/torch_stable.html
'
https://pytorch.org/get-started/previous-versions/
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
위에 없다면 여기서 찾아준다
(자신의 cuda 버전이 없다면 더 낮은 버전을 설치해야한다)
CUDA
여기서 10.2 version을 다운로드 받는다(받지 않아도 cuda를 사용할 수 있다고 뜨지만 혹시 모르니 나는 받는다..)
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
이곳에 들어가서 cu102 이니 Cuda 10.2 버전을 설치한다.
local 한번에 받기
network - 나누어 받기
특별한 이유가 없다면 한번에 받는 것이 좋다
CuDNN
https://developer.nvidia.com/rdp/cudnn-archive
[cuDNN Archive
NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.
developer.nvidia.com](https://developer.nvidia.com/rdp/cudnn-archive)
버전이 낮은게 안정적이라고 들어서 7.6.5v 을 설치해준다
다운을 받으면 이렇게 되어 있는데
안의 내용을 같은 파일 위치에 넣어준다.
이렇게 하면 설정 끝! 처음에 버전을 마음대로 깔아서 계속 다시 깔고 다시 깔고를 반복했다.. 힘들군
'Python > numpy & Pytorch' 카테고리의 다른 글
| [Pytorch] Tensor shuffle, 텐서 랜덤 섞기 (0) | 2023.03.16 |
|---|---|
| [Pytorch] tensor 합치기는 방법 cat(), stack() (0) | 2023.03.14 |
| [Pytorch] torch 설정, 랜덤 tensor 생성 (0) | 2023.03.14 |
| [Numpy] random 서브 모듈 (0) | 2022.08.01 |
| [Numpy] 데이터 생성 (0) | 2022.07.30 |
