Skip to content

Commit a2680e8

Browse files
MMathisLabAlexEMG
MMathisLab
and
AlexEMG
authored
Update installation.md (#2994)
* Update installation.md - minor edit for suggested CUDA/pytroch * Update installation.md --------- Co-authored-by: Alexander Mathis <alexander@deeplabcut.org>
1 parent 843f1a2 commit a2680e8

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

‎docs/installation.md‎

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,16 @@ conda create -n DEEPLABCUT python=3.10
1919
conda activate DEEPLABCUT
2020
conda install -c conda-forge pytables==3.8.0
2121
22-
# install torch with your desired CUDA version (or CPU) - check their website
23-
# for the exact command
24-
pip install torch torchvision
22+
# install PyTorch with your desired CUDA version (or for CPU only) - check [their
23+
](https://pytorch.org/get-started/locally/) website:
24+
# GPU version of pytorch for CUDA 11.3
25+
conda install pytorch cudatoolkit=11.3 -c pytorch
26+
2527
2628
# install the latest version of DeepLabCut
2729
pip install --pre deeplabcut
2830
# or if you want to use the GUI
29-
pip install deeplabcut[gui]
31+
pip install --pre deeplabcut[gui]
3032
3133
# ONLY IF YOU HAVE A CUDA GPU - check that PyTorch can access your GPU; this
3234
# should print `True`

0 commit comments

Comments
 (0)