安装显卡驱动:
首先添加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
先去官网https://developer.nvidia.com/cuda-downloads下载相应版本
默认打开是9.2版本,如需其他版本,可点击右下角Legacy Releases选择之前的版本
这里我选择的是cuda9.1版本,途径是runfile(local)本地安装cuda_9.1.85_387.26_linux.run.run文件
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为自己安装版本即可
在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*
安装之前,建议安装Anaconda软件,官网https://www.anaconda.com/download/下载相应版本安装即可
sudo sh anaconda.sh
打开官网https://pytorch.org/,可选择不同方式进行安装,推荐使用conda或者pip安装,安装时要注意选择cuda版本,如没有显卡,未安装cuda,则选择None,然后根据下方相应命令安装即可
默认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
更新后再重新安装即可