安装CUDA10.1、CUDNN、Pytorch

最新推荐文章于 2024-09-10 15:46:49 发布

gonetoosoon

最新推荐文章于 2024-09-10 15:46:49 发布

阅读量9.3k

点赞数

文章标签：安装 CUDA Pytorch

本文链接：https://blog.csdn.net/weixin_42089273/article/details/103171756

版权

本文详细介绍了如何在Linux环境下安装CUDA10.1、CUDNN以及Pytorch。首先，根据官方文档和显卡驱动版本选择合适的CUDA版本，并通过.run文件进行安装。接着，下载并解压CUDNN安装文件，设置环境变量确认安装成功。最后，通过Anaconda安装Pytorch，并提供Pytorch的Command命令及验证安装的步骤。

摘要由CSDN通过智能技术生成

安装CUDA10.1、CUDNN、Pytorch

如下官网链接说明了CUDA版本与NVIDIA Driver的对应关系：
CUDA Toolkit Documentation.
或参考显卡驱动的Release Notes，其中会有显卡能支持的CUDA最高版本。

如需要查询自己机器的显卡驱动版本，查询链接如下：
NVIDIA驱动下载.
也可通过
sudo apt-get install nvidia-xxx,按Tab查询nvidia显卡驱动版本
lspci | grep -i nvidia ，会显示出你的NVIDIA GPU版本信息，不过不是很详细。

1.安装CUDA10.1 .run文件

执行如下命令：

sudo sh cuda_10.1.168_418.67_linux.run --no opengl-libs

在这里插入图片描述

设置CUDA环境变量
输入 sudo gedit ~/.bashrc 打开主目录下的 .bashrc文件添加如下路径:

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