Ubuntu16.04 anaconda pytorch-gpu环境

Ubuntu16.04 anaconda pytorch-gpu环境

• 安装anaconda

• 安装nvidia显卡驱动

• 安装cuda

• 安装pytorch

---

1. 安装anaconda

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是可以加镜像源的，只能说分开一个一个下成功概率高一点
: