查找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