CUDA version : 11.0

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run

cuDNN : 8.0.5

sudo cp ./cuda/include/* /usr/local/cuda-11.0/include

sudo cp -P ./cuda/lib64/* /usr/local/cuda-11.0/lib64

sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

Pytorch version

# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

 

NVIDIA driver install

https://codechacha.com/ko/install-nvidia-driver-ubuntu/

 

우분투 18.04 - NVIDIA 드라이버를 설치하는 방법

Ubuntu 18.04에서 nvidia driver 설치하는 방법을 소개합니다. 그래픽 드라이버를 설치하는 방법은 ubuntu-drivers를 이용하여 자동으로 설치하는 방법과 드라이버 파일을 직접 다운받아 수동으로 설치하

codechacha.com

 

'구글링 저장' 카테고리의 다른 글

Quest2 - PC Unity Scene Play  (0) 2021.11.02
CUDA 설치  (0) 2021.05.31

+ Recent posts