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)

THCudaCheck失败文件= torch / csrc / cuda / Module.cpp line = 32 error = 30：   未知错误回溯（最近一次调用最近）：文件“ main.py”，   74行，在       algo = GANTrainer（output_dir）文件“ /home/anwar/Documents/StackGAN-Pytorch-master/code/trainer.py”，行   45，在 init 中       torch.cuda.set_device（self.gpus [0]）文件“ /home/anwar/.conda/envs/tensorflow/lib/python2.7/site-packages/torch/cuda/ init 。 py”，   set_device中的第262行       火炬._C._cuda_setDevice（设备）

RuntimeError：CUDA运行时错误（30）：未知错误   火炬/csrc/cuda/Module.cpp:32

Answer 1

也许您没有cudnn。记住要安装正确的版本！其次，CUDA的版本可能与pytorch所需的版本不同。我遇到了同样的错误，现在正确了。祝你好运！

RuntimeError：cuda运行时错误（30）：火炬/csrc/cuda/Module.cpp:32时出现未知错误

1 个答案: