注册教程
1、下载安装CUDA
访问英伟达官网下载CUDA,链接为https://developer.nvidia.com/cuda-downloads 。注意请下载runfile,这个比较稳妥。CUDA版本请选择10.1.如果您还想装tensorflow,那么请您装CUDA10.0。
下载安装指令如下所示:
#下载 wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run #安装 sudo sh cuda_10.1.243_418.87.00_linux.run
许多教程里说先装驱动再安装CUDA。而CUDA里自带的驱动其实也能用(只要显卡是10+以上的版本),索性一并安装了。
敲完上面代码后会跳出来一个readme。请按住空格。不要按回车键。直到出现下面的内容。然后会有一个选择安装内容的选项。“+”号就是安装。“-”号就是不安装。全部安装吧!然后会出现类似下面的内容。我这里列举的是CUDA9.0的内容。CUDA10.1与之类似。
Do you accept the previously read EULA? accept/decline/quit: accept Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81? (y)es/(n)o/(q)uit: y Install the CUDA 9.0 Toolkit? (y)es/(n)o/(q)uit: y Enter Toolkit Location [ default is /usr/local/cuda-9.0 ]: Do you want to install a symbolic link at /usr/local/cuda? (y)es/(n)o/(q)uit: y Install the CUDA 9.0 Samples? (y)es/(n)o/(q)uit: y Enter CUDA Samples Location [ default is /home/pertor ]: Installing the CUDA Toolkit in /usr/local/cuda-9.0 ... Missing recommended library: libXmu.so
装完了之后需要配置环境变量。
切记!!!要配置环境变量。否则系统找不到相关路径。打开主目录下的 .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
终端运行:source ~/.bashrc
检查:nvcc --version,如果显示nvcc版本就说明安装成功了。
2、安装cudnn
cudnn就是一个神经网络加速包,也谈不上安装,把包复制进去就好了。
(1)下载安装文件
按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive
先选中10.1版本,挑第三个cuDNN Library for Linux下载。
(2)安装cudnn
解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:
sudo cp cuda/include/cudnn.h /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*
这样,CUDA环境就安装成功了。
三、安装Pytorch以及相关包
下面要到Pytorch官网下载相应的安装包。现在的稳定版为1.3版本。直接用pip进行安装吧!
命令如下:
sudo pip3 install torch torchvision
当然,网速肯定是非常差的。怎么办呢?可以设置临时的Pip源
命令如下:
sudo pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/
你会发现运行如飞。
下面要对是否安装成功进行测试。
首先在终端敲一下:
python3
进入python命令行后,执行以下命令。如果不报错就说明安装成功了。
import torch x = torch.empty(5, 3) print(x)
如果报错,十之八九是环境变量没配置好。请在下方留言,把相关的错误代码贴上来。我们一起分析分析。
至此Pytorch安装成功。