yuta-hi / pytorch_bayesian_unet

Migrate to PyTorch. Re-implementation of Bayesian Convolutional Neural Networks (BCNNs)
MIT License
57 stars 16 forks source link

Issue with setup.py #3

Open amsinha9 opened 3 years ago

amsinha9 commented 3 years ago

I am trying to install of the necessary requirements to try out your code, however, when I try to run setup.py install, I get the error: error: Setup script exited with error: package directory 'torch/cuda' does not exist.

Do you know how I can resolve this?