0
点赞
收藏
分享

微信扫一扫

conda安装GPU版pytorch,却是cpu版本的原因

书坊尚 2022-03-24 阅读 227

原因

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

卸载

conda uninstall pytorch-mutex

安装

重新安装后,问题解决。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在这里插入图片描述

举报

相关推荐

0 条评论