cuda+cudnn+pytorch gpu

前提：按照本流程配置好了环境之后，测试代码报如下错误：

搜索之后发现是因为显卡是RTX 3090，而cuda版本太低，需要11以上版本的cuda。所以卸载cuda之后重新配置。

安装cuda+cudnn

官网下载安装包




安装成功后显示summary如下（若不成功可能是因为gcc版本太高，需要降级）：

配置环境变量：
sudo gedit ~/.bashrc
在最后添加如下：
export PATH=/usr/local/cuda-11.3/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
执行：source ~/.bashrc
查看cuda版本方法：nvcc -V

安装cudnn

官网找到对应的版本（这里我没有找到cuda11.3的，但是这个11.X的下载之后发现名字就是11.3的）：

解压下载好的压缩包：
tar xvf XXX.tgz
会解压出一个名叫cuda的文件夹。
将如下文件从解压好的文件夹中复制到cuda文件夹（注意，这里的cuda文件夹是在安装好cuda11.3之后自动出现的，和cuda-11.3在同一级目录下，不过好像复制到cuda-11.3也一样…）

sudo cp cuda/include/cudnn* /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*
查看cudnn版本方法：
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

安装pytorch

先创建一个虚拟环境，并进入。
在网上搜了很多直接用canda install的方法，最后安装好的都是cpu版本的，不知道咋回事。
所以俺直接在官网下载whl文件，然后在虚拟环境里pip instll xxx.whl进行安装。
可以在这里查看应该下什么版本：


然后在这个链接里找到对应的文件下载就行了。

然后在虚拟环境里pip instll xxx.whl进行安装。