运行时错误:cuda 运行时错误 (30):torch/csrc/cuda/模块.cpp:32 上的未知错误

Question

using config:
'CONFIG_NAME': 'stageI',
'CUDA': True,
'DATASET_NAME': 'coco',
'DATA_DIR': '/home/anwar/Documents/StackGAN-Pytorch- 
master/data/coco',
'EMBEDDING_TYPE': 'cnn-rnn',
'GAN': {'CONDITION_DIM': 128, 'DF_DIM': 96, 'GF_DIM': 192, 'R_NUM': 
4},
'GPU_ID': '1',
'IMSIZE': 64,
'NET_D': '',
'NET_G': '',
'STAGE': 1,
'STAGE1_G': '',
'TEXT': {'DIMENSION': 1024},
'TRAIN': {'BATCH_SIZE': 128,
'COEFF': {'KL': 2.0},
'DISCRIMINATOR_LR': 0.0002,
'FLAG': True,
'GENERATOR_LR': 0.0002,
'LR_DECAY_EPOCH': 20,
'MAX_EPOCH': 120,
'PRETRAINED_EPOCH': 600,
'PRETRAINED_MODEL': '',
'SNAPSHOT_INTERVAL': 10},
'VIS_COUNT': 64,
'WORKERS': 4,
'Z_DIM': 100
Load filenames from: /home/anwar/Documents/StackGAN-Pytorch- 
master/data/coco/train/filenames.pickle (82783)
embeddings:  (82783, 5, 1024)

THCuda检查失败文件=火炬/csrc/cuda/模块.cpp行=32错误=30 ： 未知错误回溯(最近一次调用(：文件"main.py"， 第 74 行，在 algo = GANTrainer(output_dir( 文件 "/home/anwar/Documents/StackGAN-Pytorch-master/code/trainer.py"，行 45， in init torch.cuda.set_device(self.gpus[0]( file "/home/anwar/.conda/envs/tensorflow/lib/python2.7/site-packages/torch/cuda/init.py"，262行，set_device torch._C._cuda_setDevice(device(

运行时错误： cuda 运行时错误 (30( ： 未知错误 火炬/csrc/cuda/模块.cpp：32

Answer 1

也许你没有cudnn。修复安装正确的版本！ 其次，CUDA 的版本可能与 pytorch 所需的版本不同。 我得到了同样的错误，现在它是正确的。 祝你好运！