最近新买了一块显卡,准备配个pytorch环境,网上教程一大堆,能用的、清晰明了的没几个,被迫重装了几次系统才搞定,在此记录一下,以防下次用到。
全新ubuntu18准备工作:
1、更换阿里源:https://blog.csdn.net/once_pluto/article/details/83384866(不然很多软件装得很慢或装不了)
2、安装 build-essential(它能帮你装好gcc、g++,后面会用到)
sudo apt-get install build-essential
下面进入正题,分四步走:
一、首先是Nidia驱动的配置
参考的教程为:https://blog.csdn.net/new_delete_/article/details/81544438
这个教程我只用了“0. 准备工作”和“1. 禁用nouveau”,后面禁用X-window服务对我有点问题(顺序不对,要先进入命令行再关闭图像界面)。
完成上面0和1后,
然后从下面这个博客的“2. 在英伟达的官网上查找你自己电脑的显卡型号然后下载相应的驱动”开始完成后面的部分。
https://blog.csdn.net/Guangli_R/article/details/86636923
最后用“nvidia-smi”测试一下没问题就行了
nvidia-smi
二、cuda + cudn的配置
这部分的参考教程是:https://blog.csdn.net/silver1225/article/details/100393719
从“搭建深度学习环境 CUDA 10.0安装”开始到“安装cuDNN”部分的结尾(也就是“安装Python、Anaconda和Pip”之前)。
三、anaconda的安装
参考教程:https://blog.csdn.net/qq_15192373/article/details/81091098
安装完成后,给 conda 更换清华源。
四、pytorch的安装
参考教程:https://blog.csdn.net/zzq060143/article/details/88042075
至此环境就搭建完成了,测试一下: