ubuntu 16.04 pytorch安装总结
版权申诉
131 浏览量
2021-01-20
03:18:27
上传
评论 1
收藏 364KB PDF 举报
ubuntu 16.04 pytorch安装总结安装总结
昨天装完了ubuntu 16.04，之前毕业设计用的tensorflow，但深度学习这门课的作业项目不大，就打算装一下Pytorch，这篇文章总结一下。主要参考了这篇博客：ubuntu16.04 +
Anaconda + CUDA10 + cudnn + pytorch + pycharm
1. 创建单独的创建单独的Anaconda环境！！环境！！
单独的Python环境真的比较重要，我这里创建了python 3.6的环境pytorch
2. 安装显卡驱动安装显卡驱动
这一步应该在安装完Ubuntu 16.04之后就已经完成了，我的上一篇博客中的方法比较复杂，还有一种方法就是从官网下载之后直接从system settings -> software & updates ->
additional driver中找到相应的驱动软件即可。
检查显卡驱动检查显卡驱动
nvidia-smi
(如果安装完pytorch之后该语句说找不到nvidia command，这时候重启就行！！！重启就行！！！)
3. 安装安装CUDA 10.0
因为参考博客上安装的是CUDA 10.0，我也就直接照葫芦画瓢了。进入CUDA 10.0对应链接
官网已经给出了安装指令，如下图：
4. 安装与安装与CUDA 10.0版本对应的版本对应的Cudnn
进入Cudnn的链接（需要帐号密码登录），然后下载
进入官方教程，将解压后include中的文件copy到CUDA文件目录下：
评论0
最新资源