계속 windows에서만 작업하다가 ubuntu에서 작업할 일이 생겨서
부랴부랴 구글의 도움을 받아 세팅을 완료했다.
이전에도 ubuntu에서 작업했지만 좀 오래되서 다 까먹음..
서버를 제공받은터라 putty를 통해 세팅을 진행했다.
또한 cuda 10.1과 cuDNN 7.6.5로 version을 정한 이유는
google colab pro 및 pytorch를 사용중인데
거기서 사용하는 version이기 때문이다. (구글 형님 믿고 갑니다)
ubuntu는 설치되어있다고 가정하에 cuda 설치를 진행한다.
cuda 설치시 driver가 설치되기 때문에 따로 graphic driver를 설치하지 않아도 된다.
- 기존 driver 및 cuda가 설치되어 있을 시 제거
$sudo apt-get purge nvidia* && sudo apt-get autoremove && sudo apt-get autoclean && sudo rm -rf /usr/local/cuda*
- cuda 10.1 update 2 다운로드 및 설치
위의 링크로 가면 Installation Instructions에 따라 다운로드 받고 설치할 수 있다.
- 설치가 잘 됐는지 version check
nvidia-smi or nvcc --version
- cuDNN 7.6.5 설치
nvidia 계정을 생성 및 로그인을 통해 cuDNN 7.6.5를 다운받자.
linux용으로 확장자가 solitairetheme8 라는 이름으로 되어있다.
아래 명령어로 압축을 해제하자.
$tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.solitairetheme8
그리고 아래의 명령어로 cudnn을 등록?하자.
$sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
- anaconda 설치
www.anaconda.com/products/individual
위의 링크에서 linux용으로 다운로드 받고 bash쉘로 실행시켜서 설치하자.
$bash ./Anaconda3-2020.07-Linux-x86_64.sh
- pytorch 설치
cuda 10.1용 pytorch를 설치하자.
$conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
- python을 실행해서 torch에서 cuda 설치가 잘 됐는지 확인하자.
import torch
torch.cuda.is_available()
설치에 이상이 없으면 True 값이 반환될 것이다.
'머신러닝 > Deep Learning' 카테고리의 다른 글
Pycharm Pro를 활용한 SSH interpreter (0) | 2020.11.23 |
---|---|
split-folders (0) | 2020.11.23 |
R-CNN(Regions with CNN features), SPPNet(Spatial Pyramid Pooling in Deep CNN) (0) | 2020.09.16 |