torch+cuda 是PyTorch深度学习框架的特定版本,并提供了相关的功能和性能优化。
一、torch-2.1.1+cu118下载地址:
https://mirrors.aliyun.com/pytorch-wheels/cu118/torch-2.1.1%2Bcu118-cp310-cp310-win_amd64.whl
https://mirror.sjtu.edu.cn/pytorch-wheels/cu118/torch-2.1.1%2Bcu118-cp310-cp310-win_amd64.whl
二、torch-2.1.1+cu121下载地址:
https://mirrors.aliyun.com/pytorch-wheels/cu121/torch-2.1.1%2Bcu121-cp310-cp310-win_amd64.whl
https://mirror.sjtu.edu.cn/pytorch-wheels/cu121/torch-2.1.1%2Bcu121-cp310-cp310-win_amd64.whl
上面2个版本任意选择一个。可以通过 cmd 输入nvidia-smi 查看自己电脑适合的版本,没有nvidia-smi也没关系。随便下载一个
- 下载完毕后。把 torch-2.1.1+cu118-cp310-cp310-win_amd64.whl
- 文件放在 小蜗安装目录中的lib -> python-3.10.11->wheel_file目录中。
- 找不到的话 可以在桌面右键小蜗图标,打开程序目录。
谨记文件名一定是 torch-2.1.1+cu118-cp310-cp310-win_amd64.whl 或 torch-2.1.1+cu121-cp310-cp310-win_amd64.whl,否则找不到。文件大小2.5G左右
目录路径:小蜗安装目录 xiaowo\lib\python-3.10.11\wheel_file
1
- 1
前往