最近换了台新的linux的ubuntu的服务器,发现其实际安装的cuda版本为11.4,但是pytorch官方给出的针对cuda 11.4并没有具体的pytorch的安装指令,于是采用不指定pytorch版本直接安装让其自动搜索得到即可
直接通过:
pip3 install torch torchvision torchaudio
安装即可
对于这种找不到和pytorch对应版本的cuda安装指令,可以试试直接只用命令来装,让他自己找版本,这个问题浪费了我周末一天半的时间,血泪。。。
最近换了台新的linux的ubuntu的服务器,发现其实际安装的cuda版本为11.4,但是pytorch官方给出的针对cuda 11.4并没有具体的pytorch的安装指令,于是采用不指定pytorch版本直接安装让其自动搜索得到即可
直接通过:
pip3 install torch torchvision torchaudio
安装即可
对于这种找不到和pytorch对应版本的cuda安装指令,可以试试直接只用命令来装,让他自己找版本,这个问题浪费了我周末一天半的时间,血泪。。。