diff --git a/docs/changes/172.maintenance.rst b/docs/changes/172.maintenance.rst
new file mode 100644
index 00000000..c94ad3d6
--- /dev/null
+++ b/docs/changes/172.maintenance.rst
@@ -0,0 +1 @@
+- Update environment to use GPU version of PyTorch
diff --git a/environment.yml b/environment.yml
index 7759585f..381508f4 100644
--- a/environment.yml
+++ b/environment.yml
@@ -5,14 +5,17 @@ channels:
     - pytorch
     - defaults
     - conda-forge
+    - nvidia
 dependencies:
   - python
-  - pytorch
+  - pytorch-gpu
+  - pytorch-cuda=12.1
+  - torchvision
   - cudatoolkit
   - cartopy
   - numpy
   - numba
+  - towncrier
   - pip
   - pip:
-    - towncrier
     - -e .