原因
用官网的命令conda安装GPU版本pytorch,发现是安装的cpu版本。查看了下库的信息,发现有一个pytorch-mutex库,这个就是原因所在,将其卸载后,再重新安装GPU版本就解决了。
卸载
conda uninstall pytorch-mutex
安装
重新安装后,问题解决。
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
微信扫一扫
用官网的命令conda安装GPU版本pytorch,发现是安装的cpu版本。查看了下库的信息,发现有一个pytorch-mutex库,这个就是原因所在,将其卸载后,再重新安装GPU版本就解决了。
conda uninstall pytorch-mutex
重新安装后,问题解决。
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
相关推荐