Ubuntu18.04 下安装CUDA,cuDNN及pytorch-gpu版本过程

第一步:

安装显卡驱动:

首先添加ppa源

sudo add-apt-repository ppa:graphics-drivers/ppa

更新一下

sudo apt-get update

安装驱动

友情提示:如果BIOS有开启Secure Boot,建议先关闭再进行安装,否则可能出现秘钥验证问题

sudo apt-get install nvidia-390

安装成功,重启系统

sudo reboot

重启系统后验证是否安装成功

nvidia-smi  

如果输入后有显示显卡信息,则安装成功

如果之后确保系统稳定,禁止更新该驱动即可

sudo apt-mark hold nvidia-390 

 

 

 

第二步:

 

 

 

cuda安装:

先去官网https://developer.nvidia.com/cuda-downloads下载相应版本

Ubuntu18.04 下安装CUDA,cuDNN及pytorch-gpu版本过程_第1张图片

默认打开是9.2版本,如需其他版本,可点击右下角Legacy Releases选择之前的版本

Ubuntu18.04 下安装CUDA,cuDNN及pytorch-gpu版本过程_第2张图片

这里我选择的是cuda9.1版本,途径是runfile(local)本地安装cuda_9.1.85_387.26_linux.run.run文件

Ubuntu18.04 下安装CUDA,cuDNN及pytorch-gpu版本过程_第3张图片

 

Ubuntu18.04默认安装的是gcc7.3,g++7.3,可通过输入以下代码查看版本

gcc --version 
g++ --version

cuda支持的版本很低,这里首先对gcc,g++降级

下载安装4.8版本

sudo apt-get install gcc-4.8  
sudo apt-get install g++-4.8 
cd /usr/bin 
sudo rm gcc
sudo ln -s gcc-4.8 gcc
sudo rm g++
sudo ln -s g++-4.8 g++

 

查看版本,如提示版本为4.8则成功

 

gcc --version
g++ --version

 

安装cuda:

使用sudo sh 路径+文件名命令进行安装    如果在主文件夹下可直接使用(可把该文件从downloads文件夹移动到主目录下)

sudo sh cuda_9.1.85_387.26_linux.run
sudo sh /home/用户名/cuda_9.1.85_387.26_linux.run.run

安装时需要注意,有一个提示是否安装显卡驱动,因为之前已经安装显卡驱动,所以选择no

安装成功后,配置环境

sudo gedit ~/.bashrc 

 

在文件的最后添加

 

  export PATH=/usr/local/cuda-9.1/bin${PATH:+:$PATH}}    
  export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  

如果安装的不是CUDA9.1版本,更改cuda-9.1为自己安装版本即可

第三步:

cudnn安装:

在https://developer.nvidia.com/rdp/cudnn-archive下载相应版本

下载之前需要注册或登录nvidia账号,之后会填一个调查问卷,然后即可下载

下载后解压文件,可以右键提取文件,也可使用命令行进行操作

tar -zxvf cudnn.tgz  

解压下载的文件,然后将解压的文件复制到cuda目录下

用户名为自己的账户名称,因解压路径不同,文件路径可能不相同,建议直接解压到主文件夹下,方便操作

sudo cp /home/用户名/cuda/targets/ppc64le-linux/include/cudnn.h /usr/local/cuda-9.1/include     

sudo cp /home/用户名/cuda/targets/ppc64le-linux/lib/libcudnn* /usr/local/cuda-9.1/lib64  

sudo chmod a+r /usr/local/cuda-9.1/include/cudnn.h /usr/local/cuda-9.1/lib64/libcudnn*

 

第四步:

 

安装pytorch:

安装之前,建议安装Anaconda软件,官网https://www.anaconda.com/download/下载相应版本安装即可

sudo sh anaconda.sh 

 

打开官网https://pytorch.org/,可选择不同方式进行安装,推荐使用conda或者pip安装,安装时要注意选择cuda版本,如没有显卡,未安装cuda,则选择None,然后根据下方相应命令安装即可

 

Ubuntu18.04 下安装CUDA,cuDNN及pytorch-gpu版本过程_第4张图片

默认pytorch0.4版本,如选择其他版本点击右下角Click here for previousversions of PyTorch 选择其他版本即可

conda安装pytorch cuda9.1版本的命令

conda install pytorch torchvision cuda91 -c pytorch

pip安装pytorch cuda9.1版本的命令

pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-linux_x86_64.whl 
pip3 install torchvision

如果使用pip安装中出现提示要求升级pip  升级即可

pip install --upgrade pip

或者,使用python更新

python -m pip install --upgrade pip

更新后再重新安装即可

 

你可能感兴趣的:(深度学习)