环境
系统：ubuntu18.04
显卡：gtx1070

之前，一直用pytorch用的好好的，今天突然不能调用gpu了，很奇怪，在网上好一通找。大部分都是说什么pytorch和cuda的版本对应不正确，我寻思不对啊之前都用的好好的怎么会突然版本不对呢？

赶紧查查cuda版本：

nvcc –V

或者

cat /usr/local/cuda/version.txt

显示是没变化的。难道是pytorch版本不小心升级了？

再去anaconda中新建一个环境，安装pytorch对应版本：

可以在官网找到合适的版本

https://pytorch.org/get-started/locally/

直接开个终端在终端用命令行安装或者在anaconda的可视化界面中直接搜索安装，它会自动安装合适版本。

可还是没有效果没办法，重新安装了显卡驱动，

ubuntu-drivers devices#查看显卡型号和推荐安装驱动

如果就要安装推荐版本，就直接运行下面的命令：

sudo ubuntu-drivers autoinstall

或者安装指定驱动

sudo apt install nvidia-460#这个驱动版本号还要去nvidia官网去查查

安装完后重启，一切正常！