注意！！！ 同时安装pytorch和cuda时，最好去官网搜索安装命令，这样可以保证版本对应
在服务器上运行代码时遇到上述错误。首先明确一点，服务器可以安装多种cuda，虚拟环境的cuda可以和服务器的不同，使用cat /usr/local/cuda/version.txt 查看的是服务器的cuda, 并不是虚拟环境的，虚拟环境用conda list直接查看cudatoolkit cudnn的版本。
如果遇到上述报错，一定是包不兼容的问题，此时需要认真检查python，tensorflow, pytorch, cuda版本的对应（在网上搜索对应版本）。如果没有问题，那么采用重装一下试试（我就是这样解决的）
将torch cuda cudnn卸载后，使用conda install pytorch torchvision cudatoolkit=9.0 来安装
然后问题解决。
小技巧：检查pytorch和cuda可不可以兼容
python
import torch
torch.cuda.is_available()
如果是true 说明安装正确
false的话 百度搜一下攻略 如 这篇教程 比较详细