linux安装pytorch（包括cuda与cudnn）

男娘i 2023-05-30 06:04 73阅读 0赞

# 安装cuda #

先查看显卡信息，有显卡才能装cuda。

lspci | grep -i nvidia

从[官网][Link 1]下载cuda，特别注意：要选择run，不要选择deb。  
然后把cuda上传到服务器。

$ chmod +x cuda_10.1.105_418.39_linux.run
    $ sudo ./cuda_10.1.105_418.39_linux.run

然后会弹出窗口，输入accept，键盘方向键选中driver，按空格键取消驱动的选择。  
安装完成后添加环境变量

export PATH=/usr/local/cuda-10.1/bin:$PATH

cuda安装后的检查一下

nvcc -V

# cudnn的安装 #

根据cuda版本选择对应的[cudnn][]进行下载，下载完成后上传到服务器。  
![在这里插入图片描述][20191107165724912.png]  
然后解压cudnn文件，并把cudnn文件夹内的文件拷贝到cuda安装目录即可。

tar -xvf cudnn-10.1-linux-x64-v7.6.4.38.tgz
    sudo cp cuda/include/cudnn.h    /usr/local/cuda-10.1/include/
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/
    sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h   /usr/local/cuda-10.1/lib64/libcudnn*

# 安装pytorch #

在官网选择版本与系统，然后按照官网给出的命令安装即可。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzE2MjI0MA_size_16_color_FFFFFF_t_70]  
如果直接执行命令下载较慢，可以用清华源，conda的清华源添加方式可以百度，这里提供linux的添加。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes

添加清华源后安装pytorch过程中出错，尝试一下重新执行安装pytorch的命令即可。

## 验证 ##

以上安装完成后，打开python，执行以下脚本。

import torch 
    
    print(torch.cuda.is_available())

若输出true，则表示cuda可以使用了。然后开始愉快的torch之旅吧。

[Link 1]: https://developer.nvidia.com/cuda-downloads
[cudnn]: https://developer.nvidia.com/rdp/cudnn-archive
[20191107165724912.png]: https://img-blog.csdnimg.cn/20191107165724912.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzE2MjI0MA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191107173350190.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzE2MjI0MA==,size_16,color_FFFFFF,t_70