已经下载conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia和12.1的cuda,代码能跑一轮,但是一轮后就报错了这,之前出现过,更改版本后直接无法使用gpu跑了
更新 torchvision 版本
pip install --upgrade torchvision
在下载之前,有个关键步骤(血泪教训)。输入以下命令,看conda下有没有cpuonly这个包,有的话卸载,否则后续默认下载的是cpu版pytorch。
conda list
卸载后在pytorch官网上找到对应的命令进行下载。
因为我的conda是10.1版本,可以参照previous-versions。又一个坑!pytorch、torchvision、torchchaudio版本要匹配,卸载不干净就回滚回以前版本“conda list --revision”。
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch