最近出了pytorch1.0的预览版(nightly版本),好久没用pytorch了,装一个耍耍。。。
安装条件: 显卡支持cuda且驱动可以安装384以上版本
Ubuntu版本升级
如果你的版本是Ubuntu 16 的话可以选择命令升级:
第一:升级到最新的Ubuntu16.04 LTS
sudo apt-get update
sudo apt-get upgrade --yes
sudo apt-get dist-upgrade --yes
第二:升级到Ubuntu18.04
sudo do-release-upgrade
#如果不行的话,输入如下
sudo do-release-upgrade -d
第三:软件配置
在升级过过程中会有一些原先安装的软件的配置询问,都选着原来的配置即可。。
第四:重启
重启过后则升级完成。
NVIDIA显卡驱动安装
第一:检查显卡和推荐驱动:
ubuntu-drivers devices
结果如下:
可以看出我的笔记本显卡是 GTX 965M ,推荐的驱动是 390
如果没有你想安装的驱动可以添加NVIDIA的PPA:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
再查看匹配驱动: