关于ubuntu下面安装cuda不对应版本的pyTorch

最近换了台新的linux的ubuntu的服务器,发现其实际安装的cuda版本为11.4,但是pytorch官方给出的针对cuda 11.4并没有具体的pytorch的安装指令,于是采用不指定pytorch版本直接安装让其自动搜索得到即可

关于ubuntu下面安装cuda不对应版本的pyTorch_第1张图片

 直接通过:

pip3 install torch torchvision torchaudio
安装即可
对于这种找不到和pytorch对应版本的cuda安装指令,可以试试直接只用命令来装,让他自己找版本,这个问题浪费了我周末一天半的时间,血泪。。。

你可能感兴趣的:(ubuntu,pytorch,linux)