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,用来验证驱动安装成功与否

本文详细介绍了如何在Ubuntu18.04上配置PyTorch环境,包括安装显卡驱动、CUDA10.2、CUDNN7.6.5、Anaconda3,并确认Python版本为3.6.10,最后成功搭建PyTorch1.6环境。
最低0.47元/天 解锁文章
801

被折叠的 条评论
为什么被折叠?



