在Linux安裝CUDA，並在Pytorch進行CUDA訓練

Linux

 2021/09/12 

首先找尋你要下載的版本

這邊我選擇CUDA11.4的版本，接著進入https://developer.nvidia.com/cuda-downloads 下載，由於我是使用Linux，所以選擇Linux版本，再來我選擇x86_64、ubuntu、20.04、deb(local)完成後照著下方給予的指令輸入至terminal:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

下載cuDNN

先進入https://developer.nvidia.com/rdp/cudnn-archive，並選擇你要的版本，這邊我選擇： Download cuDNN v8.2.1 (June 7th, 2021), for CUDA 11.x 點擊後會出現一排東西給你選，首先先下載第二排 cuDNN Library for Linux (x86_64) 再來挑選第5, 6, 7列的資料下載（我的ubuntu版本是20.04）： - cuDNN Runtime Library for Ubuntu20.04 x86_64 (Deb) - cuDNN Developer Library for Ubuntu20.04 x86_64 (Deb) - cuDNN Code Samples and User Guide for Ubuntu20.04 x86_64 (Deb) > 有帳號的記得要登入才能下載，沒帳號的要註冊一個。

完成下載

首先先cd到Download，再來輸入以下指令： ### cudnn - tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz > 請注意後面檔案的名稱要與你下載的檔案名稱相同 - sudo cp cuda/include/cudnn.h /usr/local/cuda/include - sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 - sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

libcudnn

sudo dpkg -i libcudnn8_8.2.1.32-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.1.32-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.1.32-1+cuda11.3_amd64.deb > 請注意後面檔案的名稱要與你下載的檔案名稱相同

安裝Pytorch cuda版

首先到Pytorch官網，https://pytorch.org/get-started/locally/ 然後選擇你要下載的內容，這邊我選擇完的內容是： pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

下載完後就進入terminal做測試，輸入以下程式碼來查看是否可以使用Pytorch

train_on_gpu = torch.cuda.is_available()
if not train_on_gpu:
    print('CUDA is not available.')
else:
    print('CUDA is available!')

輸出是CUDA is not available.，就完成安裝了！

by 中和高中吳振榮

Next Post

快速冪和矩陣快速冪
Previous Post

可解釋人工智慧_explainable_ai

CATALOG

1. 首先找尋你要下載的版本
2. 下載cuDNN
1. 2.1. 完成下載
  1. 2.1.1. libcudnn
3. 安裝Pytorch cuda版
1. 3.0.0.0.1. by 中和高中吳振榮