Ubuntu16.04 GTX750安装CUDA9.0,Pytorch,Anaconda教程

Ubuntu16 GTX750安装CUDA9.0,Pytorch,Anaconda教程

安装前警告

不要使用Ubuntu18!
不要使用Ubuntu18!
不要使用Ubuntu18!
务必重装成Ubuntu16

安装NVDIA驱动

只用这一个办法,一劳永逸

https://www.linuxidc.com/Linux/2019-02/157171.htm

禁用nouveau驱动

sudo gedit /etc/modprobe.d/blacklist.conf
添加
blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u
重启后用lsmod | grep nouveau,如果没有任何输出说明禁用成功。

https://blog.csdn.net/weixin_43820996/article/details/100676292

安装CUDA9.0(适配GTX750)

下载runfile(local)的安装包

不要急于安装,一定要先将NVIDIA给出官方指导手册仔细看一下

直接在图形界面打开命令行,根据官方教程安装

https://blog.csdn.net/QLULIBIN/article/details/78714596

安装cudnn

这个库安装非常简单,只需要复制文件到指定文件夹,根据官方教程即可

https://blog.csdn.net/wsc12358/article/details/81273884

安装Anaconda

Anaconda的好处是可以创建虚拟的Python环境,这意味着你不需要改Ubuntu原装的Python,自己在Anaconda怎安装卸载都行,安装后添加国内源

创建环境

conda create -n mypytorch python=3.7

激活环境

source activate mypytorch

安装Pytorch对应版本

这一步是比较容易出错的,主要原因是conda安装Pytorch总是下载超时,因此必须下载好本地文件安装

参考Pytorch官方文档https://pytorch.org/get-started/previous-versions/

CUDA9.0支持的最晚版本的Pytorch1.1

Linux and Windows CUDA 9.0

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

这个命令应该是装不上的,因为pytorch下载会超时

下载本地文件

清华镜像源下载对应版本的pytorch

文件名是pytorch-1.1.0-py3.7_cuda9.0.176_cudnn7.5.1_0.tar.bz2,torchvision-0.3.0-py37_cu9.0.176_1.tar.bz2

conda本地安装

conda install --use-local path/pytorch-1.1.0-py3.7_cuda9.0.176_cudnn7.5.1_0.tar.bz2

再次执行Pytorch的官方命令

此时,conda会自动跳过已安装的包

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0(-c pytorch 可能要去掉,不记得了)

就是因为容易安装失败,所以Anaconda可以保证我们无限次失败

https://www.jianshu.com/p/76bbf1e5ff70 这篇博客帮了大忙,感谢!

猜你喜欢

转载自www.cnblogs.com/roastpiglet/p/12074938.html