查找torch和torchvision对应版本

1、去这个网站查找:https://pytorch.org/get-started/previous-versions/

2、参考文章:https://blog.csdn.net/2302_76726543/article/details/134722054
最近训练模型跑代码需要用到nvidia的cuda架构加速,结果网上几乎找不到什么能直接解决问题的教程,最后东拼西凑了几个小时才搭建完成,所以想整理出这篇集百家之精华的教程,防止自己以后太久不用忘记了。

首先的先知道配置好这些环境需要准备好哪些东西:

  • 带有英伟达显卡的电脑(不是英伟达的也不会来找cuda吧,hh)
  • gpu版的pytorch(pip直接下的都是cpu版的,所以用不了)!!!!
  • 适配显卡的cuda架构

3、记一下本次环境:

gpu版本的torch和torchvision是在这里下载的(开头为cu)
https://download.pytorch.org/whl/torch_stable.html

https://blog.csdn.net/weixin_45508265/article/details/122006134

posted @ 2024-04-14 22:48  wendy_ann  阅读(3230)  评论(0编辑  收藏  举报