参考链接:点击打开链接
安装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 7
- cuda 9.1
- cudnn 7.05 for cuda 9.1
在选择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)
- 1
- 2
- 3
- 4
- 5
则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