Icedoom

icon company MeteoSwiss icon location Zurich icon location Machine Learning Scientist 🧪 Computer Vision 🤖 Computer Graphics 🕹️

Results 1 issues of Icedoom

CUDA Runtime Error: invalid argument after setting cuda device

6 comment

The following code fails when setting pyredner device: ``` import pyredner as pyredner import torch import matplotlib.pyplot as plt import os os.environ["CUDA_VISIBLE_DEVICES"]="0,1" class Neural_Renderer(torch.nn.Module): def __init__(self): super(Neural_Renderer, self).__init__() self.mesh_path =...