[Ubuntu] Cuda 셋팅

#Notice 적는칸

cuda 삭제 코드

sudo apt-get –purge remove “cublas” “cuda*”

Cuda 설치 사이트. (compute 86 에러 때문에 11.0이 아닌 11.1로 설치. 추후에 tensorflow와 pytorch 모두 설치 가능한 버전 확인할 것)

https://developer.nvidia.com/cuda-toolkit-archive

runfile(local)로 설치해준다.

sudo gedit /etc/bash.bashrc

ADD

export PATH=/usr/local/cuda-11.1/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.1/extras/CUPTI/lib64:$LD_LIBRARY_PATH ###################

source /etc/bash.bashrc

https://developer.nvidia.com/rdp/cudnn-archive

설치 후 cuda 폴더가 생성이 되는데요. cuda 폴더에 진입하게 되면 include 폴더와 lib64 폴더 및 txt파일이 하나 있습니다. include폴더안의 파일은 cuda의 include로 lib64폴더안의 파일은 cuda의 lib64로 복사해 주면 되는데요.아래와 같이 진행하시면 됩니다.

cd cuda

sudo cp include/cudnn* /usr/local/cuda-11.1/include

sudo cp lib64/libcudnn* /usr/local/cuda-11.1/lib64/

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

cat /usr/local/cuda-11.1/include/cudnn_version.h

grep CUDNN_MAJOR -A 2

설치 완료 확인

nvcc -V

python

print(torch.cuda.is_available())

torch.version

torch.version.cuda

cuda 버전에 맞는 pytorch 설치

pip uninstall torch

pytorch cuda version 확인하여 설치

단. 여기서 conda 말고 pip로 설치해줘야 오류가 덜 뜬다.

https://pytorch.org/get-started/previous-versions/

tensorflow cuda version

https://www.tensorflow.org/install/source#tested_build_configurations

pytorch cuda version

https://pytorch.org/get-started/previous-versions/