當前GPU的算力與當前版本的Pytorch依賴的CUDA算力不匹配（3080算力為8.6，而當前版本的pytorch依賴的CUDA算力僅支持3.7，5.0，6.0，7.0）
我的解決方法是重新到清華源網站上下載了pytorch，解決
鏈接是不同pytorch和gpu cuda cudnn版本匹配：
https://discuss.pytorch.org/t/gpu-compute-capability-support-for-each-pytorch-version/62434/7
測試cuda是否可用：
import torch torch.cuda.is_available() # 返回True 接着用下列代碼進一步測試 torch.zeros(1).cuda()