linux配置pytorch

Ubuntu16.04+Python3.6+Cuda9.2+Cudnn7.2+Pytorch0.4.1

本博客所有所需文件百度云链接(提取码：nl85)

一.安装双系统（win10+ubuntu16.04)

1.在win10下载Rufus(启动盘制作工具)官网下载、csdn下载
2.下载Ubuntu16.04镜像官网下载、中科大源、阿里云源、兰州大学源、浙江大学源
3.用Rufus制作好启动盘，进入BIOS，选择U盘启动。制作教程
4.选择Install Ubuntu,中间选择与win10共存（不要清除整个磁盘），直到安装完成，拔下U盘,点击立即重启

-注：如果开机启动后发现只有ubuntu，不要着急，开机后终端输入以下命令， 下次开机就能看到win10了:

$ sudo apt-get install grub

二.Ubuntu 更改下载源为国内源

1.备份原来的源

$ sudo cp /etc/apt/sources.list /etc/apt/sources_init.list

2.输入命令修改sources.list文件

$ sudo gedit /etc/apt/sources.list

3.在文件最前面添加以下条目（清华源,如果要添加其他源，就点我）：

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main 		restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse

4.更新源

apt-get update```

5.复损坏的软件包，尝试卸载出错的包，重新安装正确版本
  ```sudo apt-get -f install

三.安装NVIDIA驱动

1.去NVIDIA官网下载合适的驱动(我的是NVIDIA-Linux-x86_64-396.18)(驱动文件最好放在主文件夹下)
2.软件和更新 -> 附加驱动 -> 继续使用手动安装的驱动
3.准备工作

$ lsmod | grep nouveau

ubuntu自带的nouveau驱动会影响安装 若有输出,执行以下几步，没有的话就跳到4

$ cd /etc/modprobe.d
$ sudo touch blacklist-nouveau.conf
$ sudo gedit blacklist-nouveau.conf

在打开的文件中输入

blacklist nouveau
options nouveau modeset=0

$ sudo update-initramfs -u
$ reboot

重启后 重新从三.3开始
4.安装驱动
卸载掉原有驱动

$ sudo apt-get remove --purge nvidia-*

CTRL + ALT +F1 进入字符界面,以下两行命令需要记下来在进入前记下来
注意：CTRL + ALT + F7 是回到图形界面，不过本文从始至终不需要从字符界面回到图形界面
获取权限

$ sudo chmod a+x NVIDIA-Linux-x86_64-396.18.run

安装驱动

$  sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files

-no-opengl-files 意思是不安装opengl，如果安装的话可能会导致循环登陆！
重启

$ reboot

$ nvidia-smi

如果出现一个表格，则安装完成

四.安装python3.6并设为默认(参考来源)

$ add-apt-repository ppa:jonathonf/python-3.6#添加ppa到系统
$ apt-get update#更新
$  sudo apt-get install python3.6-dev
$  sudo apt-get install python3.6-gdbm
ls -l /usr/bin | grep python#查看python命令指向：
rm /usr/bin/python#删除原有python软连接
ln -s /usr/bin/python3.6 /usr/bin/python#建立python到python3.6新的软链接
apt-get install python3-pip#安装python3版本pip
ln -s /usr/bin/pip3 /usr/bin/pip#建立pip到pip3的软连接。

五.安装cuda9.2.148_396.37_linux

去Cuda官网下载Cuda9.2

$ sudo sh cuda9.2.148_396.37_linux.run #注意：提示安装驱动时选否
$ sudo gedit /etc/profile

添加如下内容：

export PATH=/usr/local/cuda-9.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64\$LD_LIBRARY_PATH

保存，重启

$ cd /usr/local/cuda-9.2/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery

结果如果是 result = PASS 则安装成功
-注： 若要卸载重装

$ cd /usr/local/cuda/bin
$ sudo ./uninstall_cuda_9.2.pl

六.安装Cudnn7.2.1.38

sudo tar -zxvf ./cudnn-9.2-linux-x64-v7.2.1.38.tgz#也可以手动右键解压（提取）
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #复制头文件
cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有动态文件
sudo ln -s libcudnn.so.7.2.1 libcudnn.so.7 #生成软衔接

注意这里要和自己下载的cudnn版本对应，可以在/usr/local/cuda/lib64下查看自己libcudnn的版本

sudo ln -s libcudnn.so.7 libcudnn.so #生成软链接

下面一步至关重要：

sudo ldconfig -v

必须在/usr/local/cuda/lib64/目录下执行，否则可能会报libcudnn.so.7: cannot open shared object file: No such file or directory
或者：

sudo ldconfig /usr/local/cuda/lib64

至此，Cudnn配置完成

七.安装Pytorch0.4.1

下载
官网下载
安装

$ sudo pip install torch-0.4.1-cp36-cp36m-linux_x86_64.whl

如果需要安装python3的包，可以按如下格式安装

$ sudo pip isntall numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

八.安装Pycharm

下载LInux 社区版（Community）
官网下载
安装方法