因为各种国内源都没有提供新版的 python 和 新版的 torch 的 gpu 版本。所以用国内源不管怎么安装都是 cpu 版本的,也就是使用 torch.cuda.is_available() 返回总是 false
以下是本文的操作环境:
- 获取下载地址 pytorch.org/get-started…
- 执行给出的命令
- 在执行时会由一个
2.5G的下载文件会很慢,直接将控制台提示的下载地址复制到迅雷中下载
4.下载完文件后直接使用 pip 命令安装即可
5.安装剩余两个依赖
执行命令时候要注意 torch 是否本重新安装,如果被重新安装成了其他版本是不行的。
pip3 install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124