Linux RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

paozaf · April 30, 2021, 11:16pm

@ptrblck here the gist:

gist.github.com

https://gist.github.com/pzaffino/7c3714ffe8eb867eb45b721ac4d2d808

pytorch_error.py

import torch
import torch.nn as nn
import torch.optim as optim
import torchvision.models
import torch.nn.functional as F
 
class Generator(nn.Module):
    def __init__(self, initial_features, num_channels=1, dropout=0.1):
        super(Generator, self).__init__()

This file has been truncated. show original

Installing Conda will take time, I have to install it without affecting the system python.