
pytorch
北部湾的落日
研究生
展开
-
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non*
Traceback (most recent call last): File "train_four_classes.py", line 164, in <module> best_weight_auc = eval_model(model, epoch_start, eval_loader) File "train_four_classes.py", line 54, in eval_model class_output = model(img) File "/h...原创 2020-05-29 16:53:26 · 8269 阅读 · 1 评论 -
PTX JIT compiler failed
我的服务器原始配置Tesla V100 cuda10.0 torch1.3.0+cu92torchvision0.2.0+cu92运行出现bug:RuntimeError: CUDA driver error: a PTX JIT compilation failedThe above operation failed in interpreter, with the following stack trace:然后我发现可能是我的torch版本过低,然后更新到torch1.5.0...原创 2020-05-27 15:38:52 · 3355 阅读 · 3 评论 -
pytorch 使用GPU报错 ->RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB
RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB (GPU 0; 10.92 GiB total capacity; 9.79 GiB already allocated; 539.44 MiB free; 10.28 MiB cached)本人的pytorch的版本是1.1.0,这个是我pytorch版本更新后,我已开的...原创 2019-07-11 13:08:24 · 53103 阅读 · 13 评论