Linux 安装cuda9.1、CUDN、pythorch

参考链接:点击打开链接

安装pytorch前需要安装cuda。pytorch的版本需要和cuda对应,pytorch下载链接：点击打开链接(下载pytorch时需对应你的python版本),我的python是3.6，所以选择cu91/torch-0.3.1-cp36-cp36m-linux_x86_64.whl,那么对应的cuda是9.1

下载安装cuda

无root权限centos 7cuda 9.1cudnn 7.05 for cuda 9.1

cuda下载 https://developer.nvidia.com/cuda-downloads.

在选择linux及对应的系统之后，选择 runfile(local)下载.

给文件运行权限chmod +x filename.run 然后 ./filename.run

在协议中选择同意（accept）,不安装driver installation （no）,然后再安装cuda时选择个人用户的目录，如/home/yourname/cuda91，sudo选择no，link选择no

修改个人用户的环境变量

环境变量文件 ~/.bashrc位于home/yourname/~/.bashrc(用vi ~/.bashrc编辑)

末尾添加如下语句

export PATH=$HOME/cuda91/bin:$PATH export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda91/lib64/

请注意修改之后 source ~/.bashrc 使环境变量生效

 查看cuda安装状态

nvidia-smi 查看显卡驱动运行状态nvcc -V 查看cuda-toolkit安装是否成功

3 安装cudnn

cudnn的安装,从官网下载https://developer.nvidia.com/cudnn（需要注册账号)，tar -xzvf cudnn-9.1-linux-x64-v7.tgz ，解压到cuda文件夹。这里输入你自己下载的安装包名。

拷贝过去cudnn->cuda (cuda91是个人用户的下的目录/home/yourname/cuda91)，这里自己注意路径的正确。

cp cuda/include/cudnn.h ~/cuda91/include/ cp cuda/lib64/libcudnn* ~/cuda91/lib64 chmod a+r ~/cuda91/include/cudnn.h ~/cuda91/lib64/libcudnn*

4 查看cudnn安装状态

cat ~/cuda91/include/cudnn.h | grep CUDNN_MAJOR -A5 显示：

#define CUDNN_MAJOR 7 #define CUDNN_MINOR 0 #define CUDNN_PATCHLEVEL 5 -- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) 12345

则cudnn版本为7.05.

接下来就可以安装你想要安装的的框架了。

安装pyTorch

这里以conda命令安装：

首先conda3换清华镜像:

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

接着使用conda命令安装pyTorch:

conda install pytorch torchvision cuda91 -c pytorch