Update README.md

biomedia-mira · Dec 23, 2021 · b1b32c8 · b1b32c8
1 parent 541281c
commit b1b32c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -22,10 +22,10 @@ Create and activate a Python 3 conda environment:
    conda activate chexploration
    ```
 
-Install PyTorch using conda:
+Install PyTorch using conda (for CUDA Toolkit 11.3):
 
    ```shell
-   conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
+   conda install pytorch torchvision cudatoolkit=11.3 -c pytorch
    ```
 
 ### Setup Python environment using virtualenv