怎么下载恰当cudn12.0版本的pytorvh

[复制链接]
查看1100 | 回复0 | 2023-8-23 11:39:39 | 显示全部楼层 |阅读模式
可以通过以下步调来下载恰当 CUDA 12.0 版本的 PyTorch:
   

  • 打开下令行/终端。
  • 运行以下下令:
    1. pip install torch==1.7.1+cu12 torchvision==0.9.0+cu12 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu12/torch_stable.html
    2. <button class="copy-btn"></button>
    复制代码
   这样就可以安装恰当 CUDA 12.0 的 PyTorch 了。如果在安装过程中碰到问题,可以实验更换源大概接洽 PyTorch 官方得到资助。

来源:https://blog.csdn.net/weixin_42584507/article/details/129509687
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则