diff --git a/environment.lock.yml b/environment.lock.yml
index 58394792..277fe961 100644
--- a/environment.lock.yml
+++ b/environment.lock.yml
@@ -41,12 +41,12 @@ dependencies:
   - brotlipy=0.7.0=py39hb9d737c_1005
   - bzip2=1.0.8=h7f98852_4
   - c-ares=1.18.1=h7f98852_0
-  - ca-certificates=2022.12.7=ha878542_0
+  - ca-certificates=2023.5.7=hbcca054_0
   - cachetools=5.2.0=pyhd8ed1ab_0
   - cairo=1.16.0=ha61ee94_1014
   - cartopy=0.21.1=py39h6e7ad6e_0
   - cdo=2.1.0=h15029e1_0
-  - certifi=2022.12.7=pyhd8ed1ab_0
+  - certifi=2023.5.7=pyhd8ed1ab_0
   - cf-units=3.1.1=py39h2ae25f5_2
   - cffi=1.15.1=py39he91dace_3
   - cfgv=3.3.1=pyhd8ed1ab_0
@@ -223,7 +223,7 @@ dependencies:
   - oauthlib=3.2.2=pyhd8ed1ab_0
   - openh264=2.1.1=h780b84a_0
   - openjpeg=2.5.0=h7d73246_1
-  - openssl=3.0.7=h0b41bf4_1
+  - openssl=3.1.0=hd590300_3
   - orc=1.8.1=hfdbbad2_0
   - ossuuid=1.6.2=hf484d3e_1000
   - packaging=21.3=pyhd8ed1ab_0
@@ -366,4 +366,4 @@ dependencies:
       - flufl-lock==7.1.1
       - netcdf4==1.6.3
       - python-cmethods==1.0.1
-prefix: /home/henry/miniconda3/envs/cuda-downscaling
+prefix: /home/henry/mambaforge/envs/cuda-downscaling
diff --git a/environment.txt b/environment.txt
new file mode 100644
index 00000000..48896b3b
--- /dev/null
+++ b/environment.txt
@@ -0,0 +1,33 @@
+# Channels:
+#   - pytorch
+#   - conda-forge
+absl-py
+black==22.10.0
+bottleneck
+cdo
+codetiming
+cudatoolkit=11.3.1
+dask
+filelock
+flake8==6.0.0
+iris
+knockknock
+metpy
+ml-collections
+mlflow
+netcdf4
+ninja
+pint
+pip
+pre-commit
+pytest
+python=3.9.15
+python-cdo
+pytorch=1.12.1=py3.9_cuda11.3_cudnn8.3.2_0
+shortuuid
+tensorboard
+torchaudio=0.12.1=py39_cu113
+torchvision=0.13.1=py39_cu113
+typer
+wandb
+xarray
diff --git a/environment.yml b/environment.yml
deleted file mode 100644
index 74fbb26e..00000000
--- a/environment.yml
+++ /dev/null
@@ -1,40 +0,0 @@
-name: cuda-downscaling
-channels:
-  - pytorch
-  - conda-forge
-  - defaults
-dependencies:
-  - absl-py
-  - black==22.10.0
-  - bottleneck
-  - cdo
-  - codetiming
-  - cudatoolkit=11.3.1
-  - dask
-  - filelock
-  - flake8==6.0.0
-  - iris
-  - knockknock
-  - metpy
-  - ml-collections
-  - mlflow
-  - netcdf4
-  - ninja
-  - pint
-  - pip
-  - pre-commit
-  - pytest
-  - python=3.9.15
-  - python-cdo
-  - pytorch=1.12.1=py3.9_cuda11.3_cudnn8.3.2_0
-  - shortuuid
-  - tensorboard
-  - torchaudio=0.12.1=py39_cu113
-  - torchvision=0.13.1=py39_cu113
-  - typer
-  - wandb
-  - xarray
-  - pip:
-    - flufl-lock
-    - python-cmethods
-prefix: /home/henry/miniconda3/envs/cuda-downscaling
diff --git a/pyproject.toml b/pyproject.toml
index af53dd00..baa99dd8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,6 +16,10 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
     "Operating System :: OS Independent",
 ]
+dependencies = [
+  "flufl-lock",
+  "python-cmethods"
+]
 
 [project.urls]
 "Homepage" = "https://github.com/henryaddison/score_sde_pytorch"