Ubuntu 18.04 Pytorch 설치 및 실행

설치

다음 글들을 참고해 설치하였다.

https://hooni-playground.com/903/

[DL 서버구축] 3. Ubuntu 18.04에 CUDA/cuDNN/PyTorch 설치 | Hooni's Playground

https://data-panic.tistory.com/19

[Ubuntu 18.04] PyTorch CUDA 10.0 환경 설정

 

 

과정

1. Nvidia 그래픽 드라이버 설치

2. CUDA 설치

3. cuDNN 설치

 

 

버전

 

1. Nvidia Driver, CUDA

➜  seunghyun nvidia-smi
Thu Sep 30 16:46:30 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.63.01    Driver Version: 470.63.01    CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:01:00.0  On |                  N/A |
| 29%   44C    P8    32W / 270W |    648MiB / 11177MiB |     15%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  NVIDIA GeForce ...  Off  | 00000000:03:00.0 Off |                  N/A |
|  0%   37C    P8    10W / 270W |     11MiB / 11178MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

 

2. cuDNN

➜  seunghyun cat /usr/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

 

3. Anaconda

➜  seunghyun conda -V 
conda 4.10.3

 

4. [conda 가상환경] Python, Pytorch

python                    3.6.13               h12debd9_1  
pytorch                   1.2.0           py3.6_cuda10.0.130_cudnn7.6.2_0    pytorch

 

 

 

오류 해결

1.  그래픽 카드 대신 llvmpipe (LLVM 10.0.0, 256 bits) 라고 뜨는 문제

NVidia 드라이버 설치 후 재부팅으로 해결

$ sudo apt-get install nvidia-driver-418
$ sudo apt-get install dkms nvidia-modprobe

드라이버 설치 및 재부팅 후에도 같은 문제가 발생한다면 secure boot가 활성화 되어있는지 확인, 비활성화 해줘야 함.

 

2. Package ncurses conflicts for: sqlite -> ncurses[version='>=6.2,<7.0a0'] ncurses

conda 가상환경에서 pytorch 설치 시 버전 컨플릭로 발생
일반 환경에서 설치한  pytorch uninstall 후 가상환경 shell에서 버전에 맞게 재설치

 

 

실행 확인

import torch

print(torch.cuda.get_device_name(0))
print(torch.cuda.is_available())

out

NVIDIA GeForce GTX 1080 Ti
True

Process finished with exit code 0