报错信息 一直是 InternalError: Dst tensor is not initialized.
然后显示一大堆信息,通过谷歌大部分找到的就是 GPU 显存溢出。然后
加上 CUDA_VISIBLE_DEVICES=1
Environment Variable Syntax Results
CUDA_VISIBLE_DEVICES=1 Only device 1 will be seen
CUDA_VISIBLE_DEVICES=0,1 Devices 0 and 1 will be visible
CUDA_VISIBLE_DEVICES="0,1" Same as above, quotation marks are optional
CUDA_VISIBLE_DEVICES=0,2,3 Devices 0, 2, 3 will be visible; device 1 is masked
这里注意的一点就是 CUDA_VISIBLE_DEVICES 必须大写。