本文主要是介绍【pyTorch安装版本真麻烦】,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1、pytorch 2.0.1版本可以用cuda11.7
# CUDA 11.7
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
2、pytorch 2.1 版本后最低cuda11.8
3、安装pytorch时候,注意torchaudio 版本
总之注意pyTorch安装版本真麻烦
参考官网
Previous PyTorch Versions | PyTorch
这篇关于【pyTorch安装版本真麻烦】的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!