解决安装torch后,torch.cuda.is_available()结果为false的问题
清华源下载到的torch==2.5.1版本的Lib\site-packages\torch\version.py
其中,清华源指的是:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
cuda: Optional[str] = None
看了这篇帖子解决安装torch后,torch.cuda.is_available()结果为false的问题
才知道在PyTorch可以看对应版本,PyTorch官网https://pytorch.org/
我的python是3.10的,CUDA是12.4的,驱动对应的是对应的12.6的
按理说没问题的,但是还是False,于是我卸载掉我的torch,然后执行官网上的指令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
然后下载速度很慢,我看也就是下载一个.whl文件
于是我在CMD窗口复制了下载地址用迅雷下载了。
https://download.pytorch.org/whl/cu124/torch-2.5.1%2Bcu124-cp310-cp310-win_amd64.whl
这个文件有2.34GB
下载好.whl文件以后,用虚拟环境pip install 文件名安装以后得到的torch包就可以正常运行了。
我查看了Lib\site-packages\torch\version.py
这真的太坑了!