【BUG】conda 安装PyTorch CPU换GPU版本失败

我的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,成功。

猜你喜欢

转载自blog.csdn.net/xiangduixuexi/article/details/107584577