본문 바로가기
lang/linux

[버전관리] Ubuntu - Cuda - Cudnn

by Wordbe 2019. 7. 30.
728x90

1. Ubuntu version check

$ lsb_release -a

No LSB modules are available.     
Distributor ID: Ubuntu            
Description:    Ubuntu 18.04.2 LTS
Release:        18.04             
Codename:       bionic            

2. CUDA version check

$ nvcc --version (혹은 nvcc -V)

$ locate cuda | grep /cuda$ (경로 확인)

3. Cudnn version check

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

4. Pytorch는 설치시 cudatoolkit을 제공한다. cudnn 까지 패키지로 묶어서 깔아준다.

$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
728x90

'lang > linux' 카테고리의 다른 글

리눅스 휴지통 경로  (0) 2019.08.14
리눅스 명령어 정리 (Linux commands)  (963) 2019.06.28

댓글