后端 · 2024年 3月 5日 0

python安装对应英伟达显卡的pytorch

首先安装cuda，下载安装即可，此步忽略

确定cuda版本

确定完版本后进入官网

https://pytorch.org/

然后进入所有版本的页面

Previous PyTorch Versions | PyTorch

寻找符合你电脑cuda版本的最新的符合你的平台的命令（比如windows

开启代理的增强模式（要不然时间太长

将命令在要安装的环境运行

然后就可以用torch啦

标签： python pytorch

您可能还喜欢...

发表回复取消回复