Ubuntu - (1) Pytorch, cuda 설치 및 환경변수 설정

pytorch 다운

본인 cpu와 운영체제에 맞는 버전 택하면 된다.

pytorch 공홈 다운 링크

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

cuda 다운

cuda toolkit 페이지

pytorch 버전에 맞는 cuda를 받으면 된다.

wget <https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin>
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget <https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb>
sudo dpkg -i cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

요상한 오류 뜨면 그냥 명령어 앞에 sudo를 붙이면 된다.

환경변수

vim ~/.bashrc
#또는
vim ~/.profile

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

결과 확인

nvcc -V
#또는
nvcc -version

nvcc 없다고 뜨면 오류 코드에 있는 패키지를 pip으로 설치하면 된다.