根据pytorch官网的命令安装cuda版本pytorch试了几次最后torch.cuda.is_available()都是False
排除了CUDA版本问题以后,开始疯狂试错。
最后找到原因:清华镜像下下来有3个包是CPU版本的。可以自行在pkg文件夹或者conda list找找,排除一下是否是这个原因。
解决方案:搭了个梯子,用默认源下载。但是11.8的版本默认命令无法找到torchaudio的包,所以对照GPU版的Pytorch、torch、torchvision、torchaudio安装(用pip,无conda)_pip install torch==2.1.0 torchvision==0.16.0 torch-CSDN博客找了一下我自己python版本可用的torch和torchvision包,在官网找了以前的版本
最后终于安装成功。