saic-violet / bilayer-model

Mozilla Public License 2.0
246 stars 48 forks source link

Can't run #15

Open statisticalwork opened 3 years ago

statisticalwork commented 3 years ago

Did anyone manage to run the code?

ImportError Traceback (most recent call last) /bilayer-model/examples/inference.ipynb in 8 from matplotlib import pyplot as plt 9 ---> 10 from infer import InferenceWrapper

/bilayer-model/infer.py in 1 import argparse ----> 2 import torch 3 from torch import nn 4 from torchvision import transforms 5 from PIL import Image

~/anaconda3/lib/python3.7/site-packages/torch/init.py in 100 pass 101 --> 102 from torch._C import * 103 104 all += [name for name in dir(_C)

ImportError: /root/anaconda3/lib/python3.7/site-packages/torch/lib/libtorch.so.1: undefined symbol: nvrtcGetProgramLogSize

image