Update environment to install pytorch gpu

2026-06-04 15:04:24 +08:00 · 2020-11-17 15:16:35 -05:00
parent 3f01cc3371
commit afafea8791
2 changed files with 9 additions and 5 deletions
--- a/docs/source/installation.rst
+++ b/docs/source/installation.rst
@@ -23,7 +23,7 @@ Set up environment

    $ cd D-SCRIPT

-    $ conda env create --file environment.yml # Edit this file to change CUDA and cuDNN version if necessary
+    $ conda env create --file environment.yml # Edit this file to change CUDA version if necessary

    $ conda activate dscript

--- a/environment.yml
+++ b/environment.yml
@@ -1,12 +1,16 @@
 name: dscript
+
 channels:
-  - defaults
-  - bioconda
  - pytorch
+  - conda-forge
+  - defaults
+
 dependencies:
+  - cudatoolkit=10.2
+  - pip=20.0
  - python=3.7
-  - pytorch=1.3
-  - cudatoolkit=10.0
+  - pytorch=1.5
+  - torchvision=0.6
  - h5py
  - matplotlib
  - numpy