我的PyTorch原来是CPU版本的,转到GPU版本发现没有办法下载GPU版本。具体来讲,执行PyTorch官网给的命令:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
并不能下载GPU版本的PyTorch,仍下载CPU版本的。
输入 conda list
后,发现其中有一个包叫做:
cpuonly
其中后面跟了一个pytorch字样。
如下:
将该包卸载 conda uninstall cpuonly
,
再次安装GPU版本PyTorch,成功。