- 博客(5)
- 收藏
- 关注
原创 安装pytorch踩坑记录
当时pip install vit_pytorch提示我了,得是torch1.10才能兼容。结果发现cuda不能用了,一看torch版本才发现变成最新的了,估计被自动升级了。还有就是,本来都安装好了,结果跑vit的时候安包。要根据cuda版本选择torch版本。然后我试了试作者给的代码,没报错。要重装torch,不过挺快的。
2023-01-14 21:42:53
289
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人