环境:
GPU2080ti
ubuntu 18.04
python 3.6
cuda 11.1
cudnn 8.10 for cuda 11.0, 11.1, 11.2
pytorch 对应版本
方法1: 修改batchsize
方法2:改用root用户去运行即可, 原因未知
环境:
GPU2080ti
ubuntu 18.04
python 3.6
cuda 11.1
cudnn 8.10 for cuda 11.0, 11.1, 11.2
pytorch 对应版本
方法1: 修改batchsize
方法2:改用root用户去运行即可, 原因未知