Anaconda&英偉達driver&cuda&cudnn&pytorch安裝(2020)

1. Anaconda安裝

清華鏡像下載鏈接
下載好之後 bash ~/Downloads/Anaconda3-5.2.0-Linux-x86_64.sh
一路 yes 或者 Enter 最後可以選擇不裝 VSCode， 如果中間忘記寫入環境變量，可以在 ~/.bashrc 中添加 export PATH=/home/xxx/anaconda3/bin:$PATH， xxx 這裏是你linux用戶名。

2. nvidia 驅動安裝

下載地址
首先卸載之前驅動:

sudo apt-get remove nvidia-*
sudo apt-get autoremove  # 加上才能卸載乾淨原驅動

CTRL+ALT+F1 進入文本模式
sudo service lightdm stop 關閉顯示服務
chmod a+x NVIDIA-Linux-x86_64-4xx.43.run 修改權限, 注意文件名
sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files 安裝

-no-x-check 安裝驅動時關閉x服務
-no-nouveau-check 安裝驅動時禁用Nouveau
-no-opengl-files 安裝時只裝驅動文件，不安裝Opengl

關於禁用Nouveau,還有一種方法是在/etc/modprobe.d/blacklist.conf的最後一行加上 blacklist nouveau
sudo update-initramfs -u 使其生效
lsmod | grep nouveau 使用這個命令看是否有輸出，如果沒有證明禁用成功
選項依次爲:
accept->Continue installation-> NO(nvidia-xconfig) -> NO(32-Bit)
sudo service lightdm restart 啓動顯示服務
nvidia-smi 顯示信息的話表示安裝成功

3. 安裝CUDA

下載鏈接
下載好之後:

chmod a+x cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

詢問是否安裝驅動的時候選擇 no(n)
詢問是否創建鏈接和安裝samples選擇 yes(y)

##添加環境變量##
在~/.bashrc中添加:

LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
PATH=$PATH:/usr/local/cuda-10.1/bin
CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

輸入 nvcc --version 如果有下面顯示則表示安裝成功:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

也可以通過編譯 samples 來測試是否安裝成功, 如果最後輸出PASS，則安裝成功:

cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery

4. 安裝CUDNN

下載鏈接
解壓好文件:

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看cudnn是否安裝成功, 如果輸出如下則表示安裝成功:

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"

5. 安裝pytorch

官網
新建conda 虛擬環境:

conda create --name pytorch python=3.6
conda activate pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
# 這裏也可以指定pytorch版本, 例如
conda install pytorch=1.1.0 torchvision cudatoolkit=10.1 -c pytorch

安裝好測試是否安裝成功:

>>> import torch
>>> print(torch.__version__)
1.1.0
>>> torch.cuda.is_available()
True
>>> a = torch.Tensor(5,3)
>>> a=a.cuda()
>>> print(a)
tensor([[2.8026e-44, 0.0000e+00, 1.1210e-44],
        [0.0000e+00, 2.5104e+31, 3.0927e-41],
        [3.9236e-44, 0.0000e+00, 3.9236e-44],
        [0.0000e+00, 0.0000e+00, 0.0000e+00],
        [1.3740e+32, 3.0927e-41, 2.3822e-44]], device='cuda:0')