在训练刚开始至图片位置时停止卡了快一个钟了也不动,用cpu训练秒开
环境为cuda11.8
cudatoolkit 11.6
python 3.7
pytorch 1.7.0+cu101
torchvision 0.8.1+cu101
gpu是rtx 3070ti laptop
在训练刚开始至图片位置时停止卡了快一个钟了也不动,用cpu训练秒开
环境为cuda11.8
cudatoolkit 11.6
python 3.7
pytorch 1.7.0+cu101
torchvision 0.8.1+cu101
gpu是rtx 3070ti laptop
你的pytorch的cuda版本选错了,torch1.7.0是支持cuda11.x的,但是你下载的是cu101,这个就表示支持的cuda是10.1。但是你环境里面的cuda是11.x(30系显卡不支持cuda11.0以下的版本),所以运行不了。你需要换个cuda11.0的torch就行。另外,torchvision 0.8.1+cu101这个是torch1.7.1的,不匹配会报错的。
详细的信息可以看
Previous PyTorch Versions | PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
https://pytorch.org/get-started/previous-versions/