用官网的方法安装,验证torch.cuda.is_available()总是False.
又说是因为官网上下载的总是cpu版本,还有说是因为cuda 的。具体原因不清楚。
参考https://blog.csdn.net/qq_43900762/article/details/131441360
命令:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
然后就True了