安装pytorch踩坑记录

安装pytorch踩坑记录_第1张图片

安装pytorch踩坑记录_第2张图片

要根据cuda版本选择torch版本

还有就是,本来都安装好了,结果跑vit的时候安包 

当时pip install vit_pytorch提示我了,得是torch1.10才能兼容

然后我试了试作者给的代码,没报错

pip -q install vit_pytorch linformer  -i https://pypi.tuna.tsinghua.edu.cn/simple

结果发现cuda不能用了,一看torch版本才发现变成最新的了,估计被自动升级了

要重装torch,不过挺快的

 

你可能感兴趣的:(pytorch,深度学习,python)