Ubuntu配置PyTorch记录

ubuntu重装过几次，每次都要重新配置，得甄别网上的教程，今天记录下自己的配置经过，便于以后用到。

一、安装显卡驱动

1、首先安装好ubuntu系统

2、安装nvidia驱动

先从官网下载驱动安装包

禁用ubuntu自带的显卡驱动：

#如有输出信息则需要禁用nouveau
lsmod | grep nouveau

#禁用
cd /etc/modprobe.d
sudo vi blacklist-nouveau.conf

#在文件末尾添加如下信息
blacklist nouveau
options nouveau modeset=0

#更新 重启
sudo update-initramfs -u
sudo reboot

#重启后 再次检查 
#若无信息输出 则禁用成功
lsmod | grep nouveau

安装下载的显卡驱动：

#给驱动run文件赋予执行权限：
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
#安装，后面的参数非常重要，不可省略：
sudo ./NVIDIA-Linux-x86_64-384.59.run –no-opengl-files

一路选择默认安装

安装完输入nvidia-smi，输出显卡信息则表明安装成功

二、安装cuda

下载对应版本cuda

#安装
sudo ./cuda_10.1.168_418.67_linux.run

若要卸载则进入/usr/local/cuda-10.1/bin运行cuda-uninstaller

安装完成后配置信息

sudo vi ~/.bashrc
#文件末尾添加如下信息
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64 #注意冒号

#更新配置
sudo source ~/.bashrc

检查是否安装成功

cd  /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
sudo make 
./deviceQuery
#显示关于gpu信息则安装成功

三、cudnn安装

 下载对应版本的cudnn library for linux

解压复制到如下文件夹即可完成

tar -zxvf cudnn-10.1-linux-x64-v7.6.tgz
cd cuda    
sudo cp lib64/lib* /usr/local/cuda-10.1/lib64/    
sudo cp include/cudnn.h /usr/local/cuda-10.1/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64    
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*

四、安装Pytorch

现将python版本升级为3.7，并制定更改系统默认Python版本为3.7

再pip3安装pytorch