- 博客(1)
- 收藏
- 关注
原创 使用阿里云镜像本地安装torch的cu121版本示例
我的cuda是12.1,用的是python3.10,torch版本根据自己想要的安装,但是torch,torchaudio ,torchvision版本要对应,比如下面cu121对应的是CUDA12.1,cp310-win_amd是对应的windows下的python3.10。下载完成后cd到上述文件的目录,3个文件直接使用pip进行安装,例如。输出应该是2.3.1+cu121,这和自己下载的版本是对应的。从上述网址中找torch的包,比直接从官网安装快很多。3个文件都安装完毕后在终端输入。
2024-07-25 23:25:07
838
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人