diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..61a709b --- /dev/null +++ b/environment.yml @@ -0,0 +1,14 @@ +name: ium_test +channels: + - defaults +dependencies: + - mlflow + - numpy + - pandas + - scikit-learn + - torchvision + - torchaudio + - pytorch + - pytorch-cuda=11.8 + - os +prefix: /home/witek/miniconda3/envs/ium_test