重装无数次pytorch-gpu之后的小技巧
安装GPU版本的pytorch,按照以下方式
https://blog.youkuaiyun.com/uuhhy/article/details/124638448
安装之后执行
python
import torch
print(torch.cuda.is_available())
仍然是Flase的时候,可以尝试一下这个办法,先安装一个cpuonly的包,再卸载他,pytorch-gpu的就安装好啦!
conda install cpuonly
conda uninstall cpuonly
撒花!!!