【问题标题】:RuntimeError: cuda runtime error (30) : unknown error at torch/csrc/cuda/Module.cpp:32RuntimeError:cuda 运行时错误(30):torch/csrc/cuda/Module.cpp:32 处的未知错误
【发布时间】:2019-02-24 11:39:31
【问题描述】:
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 FAIL 文件=torch/csrc/cuda/Module.cpp 行=32 错误=30: 未知错误 Traceback(最近一次调用最后一次):文件“main.py”, 第 74 行,在 算法 = GANTrainer(output_dir) 文件“/home/anwar/Documents/StackGAN-Pytorch-master/code/trainer.py”,行 45,在初始化 torch.cuda.set_device(self.gpus[0]) 文件“/home/anwar/.conda/envs/tensorflow/lib/python2.7/site-packages/torch/cuda/init. py", 第 262 行,在 set_device 中 torch._C._cuda_setDevice(设备)

RuntimeError: cuda 运行时错误 (30) : 未知错误 火炬/csrc/cuda/Module.cpp:32

【问题讨论】:

    标签: python python-2.7 tensorflow pytorch


    【解决方案1】:

    也许你没有cudnn。记住要安装正确的版本! 其次,CUDA的版本可能和pytorch需要的不同。 我得到了同样的错误,现在它是正确的。 祝你好运!

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2011-04-16
      • 1970-01-01
      • 1970-01-01
      • 2012-12-09
      • 1970-01-01
      相关资源
      最近更新 更多