torch cpu 版本与 cuda版本 安装
·
不同点：
0. 名字不同：cpu版本: cuda=2.1.0 cuda版本：cuda=2.1.0+cu121
1. 尺寸不同：cpu版本只有500多MB, cuda版本有2个G
2. 镜像不同：清华源有cpu版本，但是没有cuda版本
安装cuda版本：
0. 安装官网链接：pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121
1. 找到控制台输出的下载链接： https://download.pytorch.org/whl/cu121/torch-2.1.0%2Bcu121-cp38-cp38-win_amd64.whl
2. 点击这个链接 下载轮子。
3. pip install 轮子路径
4. ok！
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)