先直接上图,这是我最后修改之后的结果,返回值为True

看一下CUDA版本,确认为CUDA11.5

 

 现在说方法:

之前我根据官网的输入安装指令:Start Locally | PyTorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 可以看到官网给的代码是=11.3!!这个就是关键,不要去安装=11.5的版本。11.3的版本就兼容。11.5的版本在各种源里都找不到,如果你安装11.5的话pytorch就会自动安装CPU版本的。那就会返回FALSE。(当安装的时候如果是一个多G的pytorch,就是GPU版本,就是正确的。如果是几百兆就是CPU,就是错的)

 总结一下安装的版本:                ​​​​​​​                                 ​​​​​​​

 

最后再介绍一下一个anaconda的源,现在清华源又维护可以用了。

 

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/bioconda/

 

 

 

 

 

 

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐