diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000000000000000000000000000000000000..c2968a9e059f8a1d0a4b9438234e311a1bfbe168 --- /dev/null +++ b/environment.yml @@ -0,0 +1,17 @@ +name: sbi_workshop +channels: +- pytorch +- defaults +- conda-forge +dependencies: +- python=3.7.10 +- pytorch=1.8.1 +- ipykernel==5.1.3 +- ipywidgets +- scikit-learn +- scipy +- pip +- pip: + - "sbi==0.16.0" + - "rise==5.7.1" + - sbibm diff --git a/modules.sh b/modules.sh index ede574476a89436cdcb8f23cfb067956e06dd69a..e5b9f450c38f5bd960e578e0d16a8dbd2f3763a5 100644 --- a/modules.sh +++ b/modules.sh @@ -1 +1,2 @@ module purge +ml CUDA/11.3