解决torch.version.cuda=None以及torch.cuda.is_available()=False
然后就出现了上面的问题，看了下别人的博客，好像如果conda配置了清华镜像源，然后conda install安装的就是cpu版本的torch，不是gpu版本的，我也不知道是不是这个问题。先把安装的torch卸载了，然后用pip安装gpu版本的torch（我没用镜像源，我开了外网下好像），
·
我之前使用conda install命令安装gpu版本的pytorch的，如下：
然后就出现了上面的问题，看了下别人的博客，好像如果conda配置了清华镜像源，然后conda install安装的就是cpu版本的torch，不是gpu版本的，我也不知道是不是这个问题
解决办法：
先把安装的torch卸载了，然后用pip安装gpu版本的torch（我没用镜像源，我开了外网下好像），
即执行命令下面即可
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
6
0- 0
已为社区贡献1条内容
所有评论(0)