RuntimeError: CUDA 错误: 未知错误

backcall 0.1.0 py36_0
blas 1.0 mkl
ca-certificates 2019.1.23 0
certifi 2018.11.29 py36_0
cffi 1.12.1 py36h2e261b9_0
cudatoolkit 9.0 h13b8566_0
cycler 0.10.0 pypi_0 pypi cython 0.29.6 pypi_0 pypi decorator 4.3.2 py36_0
intel-openmp 2019.1 144
ipython 7.3.0 py36h39e3cac_0
ipython_genutils 0.2.0 py36_0
jedi 0.13.3 py36_0
kiwisolver 1.0.1 pypi_0 pypi libedit 3.1.20181209 hc058e9b_0
libffi 3.2.1 hd88cf55_4
libgcc-ng 8.2.0 hdf63c60_1
libgfortran-ng 7.3.0 hdf63c60_0
libstdcxx-ng 8.2.0 hdf63c60_1
maskrcnn-benchmark 0.1 dev_0 matplotlib 3.0.2 pypi_0 pypi mkl 2019.1 144
mkl_fft 1.0.10 py36ha843d7b_0
mkl_random 1.0.2 py36hd81dba3_0 …