Ubuntu16.04 anaconda pytorch-gpu环境

Ubuntu16.04 anaconda pytorch-gpu环境

安装anaconda

安装nvidia显卡驱动

安装cuda

安装pytorch

1、 安装anaconda
直接在官网下载run文件 https://www.anaconda.com/download/

文件比较大，下载速度慢，果断扔网盘 http://pan.baidu.com/s/1qYdYiz6 密码：y56i

直接 ./Anaconda3-5.0.1-Linux-x86_64.sh

2、 安装nvidia显卡驱动

先卸载掉原有驱动

sudo apt-get remove –purge nvidia*

修改/etc/modprobe.d/blacklist.conf

在最后一行添加 blacklist nouveau #禁用nouveau第三方驱动，之后也不需要改回来

sudo update-initramfs -u

按官网上的引导下载 http://www.nvidia.cn/Download/index.aspx?lang=cn



需要先关闭x界面

Ctrl + Alt + F1~F6进入命令行界面，F7显示桌面

sudo /etc/init.d/lightdm stop

安装驱动，记得加后面的选项，否则的话会导致无法登录到桌面

sudo ./NVIDIA-Linux-x86_64-384.98.run –no-opengl-files

出错了的话看看选项，不同机器不一定通用

安装成功后重启，检查nvidia-smi，有打印gpu信息的表格就说明成功了

如果没有成功，可以到BIOS里关闭Secure Boot里的UEFI，再重装一遍试试

3、 安装cuda

到官网下载

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

首选runfile，不过有的机器安装时会出现内存不足的问题，那就用deb吧，总之都ok



4、 安装pytorch
官网 http://pytorch.org/

也是一慢出翔的网站暂时没找到什么办法，conda是可以加镜像源的，只能说分开一个一个下成功概率高一点