BoChenGroup / BIRNAT

34 stars 10 forks source link

Add docker image #8

Open bryanbocao opened 1 year ago

bryanbocao commented 1 year ago

Prepare a docker image https://hub.docker.com/repository/docker/bryanbocao/birnat/general. Run the code in a docker container without reinstalling the environment. Best use if CUDA or PyTorch versions change.