linux下 pytorch gpu cuda cudnn 安装

1. 首先确认linux 是否安装驱动

因为我已经安装了，执行命令：nvidia-smi

 看到如下信息：

第一个框是驱动版本，第二个框支持的cuda最高版本

2. 确认自己要安装的torch版本和cuda版本

Previous PyTorch Versions | PyTorch   到这个网站自己找

下面这个是我自己要安装的版本，但这一步不装，下面装，只是先确定版本

# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html

3. 下载cuda和cudnn文件

具体如何下载参考这篇博客，只需要把window的对应换成linux即可

window下 torch gpu cuda cudnn 安装_泯灭XzWz的博客-CSDN博客_安装torch-gpu

上面 两个包是我下载好的样子

 4. 安装cuda

在cuda安装包目录下，执行

sudo sh cuda_11.1.1_455.32.00_linux.run

几点注意的：

1.如果你已经安装了驱动，会有如下提示：

Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.

不管他直接选择continue

2. 出现如下的界面时，如过你已经安装驱动，选中光标选中Driver,按一下 enter键，把前面的X去掉，然后选择install 回车安装。

3. 安装好后配置环境变量

vi ~/.bashrc

加入如下配置

export PATH=/usr/local/cuda-11.1/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64

 4. 验证cuda是否安装成功

按下面操作

cd /usr/local/cuda-11.1/samples/cuda-samples/Samples/1_Utilities/deviceQuery

sudo make

sudo ./deviceQuery

输入如下信息则安装成功：

5. 安装cudnn

1.解压 安装包

tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz

2. 复制并更改权限

$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

6. 安装torch

这一步就安装之前我们确认的torch版本

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

7.验证

成功！

8. 参考

CUDA Installation Guide for Linux

觉得有用，帮忙点个赞呗  >.<