RuntimeError: cuda runtime error (11) : invalid argument at /opt.../THC/THCGeneral.cpp:663

起名字什么的好难

于 2020-08-19 12:46:15 发布

阅读量1.8k

点赞数 2

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/u012428169/article/details/108098009

版权

I have installed pytorch 0.4.1 by the followint command line

conda install pytorch=0.4.1 cuda90 -c pytorch

but encounter the following error

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1535493744281/work/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument
Exception ignored in: <function _DataLoaderIter.__del__ at 0x7f4373417290>
Traceback (most recent call last):
  File "/home/yuxijin/anaconda3/envs/LinearStyle/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 399, in __del__
    self._shutdown_workers()
  File "/home/yuxijin/anaconda3/envs/LinearStyle/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 378, in _shutdown_workers
    self.worker_result_queue.get()
  File "/home/yuxijin/anaconda3/envs/LinearStyle/lib/python3.7/multiprocessing/queues.py", line 354, in get
    return _ForkingPickler.loads(res)
  File "/home/yuxijin/anacond