更换tensorflow版本为2.4.0以上,好像说是30显卡的问题(我是rtx3060)
如:
cudatoolkit=11.3 cudnn=8.2.1 tensorflow-gpu=2.5.0
安装(国内源好下载的版本):
conda install tensorflow-gpu=2.5.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
tensorflow cuda cudnn对应版本 参考:
https://tensorflow.google.cn/install/source_windows
备注:
如果在win10上 gpu训练前期等待太久,
如Adding visible gpu devices: 0卡住
高级系统设置->环境变量:
变量名:CUDA_CACHE_MAXSIZE
变量值:6442450944 (nvidia-smi查看自己的gpu显存,如RTX3060为6144MiB : 6144 * 1024 * 1024 =6442450944)
参考:TensorFlow Adding visible gpu devices: 0 加载GPU速度慢