ubuntu(18.04)配置pytorch(cuda10.2+cudnn7.6.5)环境
安装流程:
1安装显卡驱动
2安装CUDA
3安装CUDNN
4安装Anaconda3
5确定python版本
6搭建pytorch环境
7总结
1安装显卡驱动
首先查阅自己电脑的显卡配置,打开终端,
输入命令:sudo lshw -c display 回车
按要求输入用户密码
将会看到自己显卡的具体信息
输入命令:sudo ubuntu-drivers devices 回车
将会看到系统给你推荐的三款显卡驱动
输入命令:sudo ubuntu-drivers autoinstall 回车
将会自动安装上一步推荐的显卡驱动
输入命令:sudo apt-get install mesa-utils 回车
将会安装mesa-utils,用来验证驱动安装成功与否
输入命令:glxinfo | grep rendering
结果为Yes,则驱动安装完成。