原因

用官网的命令conda安装GPU版本pytorch,发现是安装的cpu版本。查看了下库的信息,发现有一个pytorch-mutex库,这个就是原因所在,将其卸载后,再重新安装GPU版本就解决了。
在这里插入图片描述

卸载

conda uninstall pytorch-mutex

安装

重新安装后,问题解决。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在这里插入图片描述

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐