Ubuntu 16.04 에서 Pytorch 설치하기 - yspaik/pytorch_study GitHub Wiki

실행환경

• Ubuntu 16.04
• GPU NVIDIA GeForce GTX 1080 Ti

1. 그래픽 드라이버 설치

• repository 대신 nvidia homepage에 가서 graphic driver를 찾아 최신판으로 설치: 410.79

• 설치 후 재부팅

2. CUDA 설치

3. CuDNN 설치

환경변수 설정  vi .bashrc

# cuda environement

export PATH=/usr/local/cuda/bin:${PATH}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}

4. Anaconda3 설치

wget https://repo.continuum.io/archive/Anaconda3-2018.12-Linux-x86_64.sh

time bash Anaconda3-2018.12-Linux-x86_64.sh

• Anaconda 설치 중 /usr/anaconda3 로 directory를 지정해서 깔기 (multi user 사용자 차원)

• vi를 열어 .bashrc 가장 아래에 아래 코드 삽입

  source /usr/anaconda3/etc/profile.d/conda.sh

  . .bashrc

• conda를 입력하여 실행이 되는지 확인

5. pytorch 설치 pytorch home page에 가면 어떤 명령어로 설치해야 하는지 알려준다.

- conda create -n pytorch python=3.7

- conda activate pytorch

- conda install pytorch torchvision -c pytorch

python version 확인 python 실행 후

>>> import torch

>>> torch.__version__

'1.0.0'