Visitar URL original
Add PyTorch🔥 documentation by n-poulsen · Pull Request #2605 · DeepLabCut/DeepLabCut · GitHub
Skip to content
Merged
Changes from 1 commit
Commits
Show all changes
63 commits
Select commit Hold shift + click to select a range
9aa9647
added initial pytorch docs
Jun 7, 2024
76cd889
Update _toc.yml
MMathisLab Jun 7, 2024
f15a09a
Update user_guide.md
MMathisLab Jun 7, 2024
f448dc9
Update architectures.md
MMathisLab Jun 7, 2024
d598a56
Update labelling.md
MMathisLab Jun 7, 2024
b9195b8
Update labelling.md
MMathisLab Jun 7, 2024
4bc8aa1
Update codespell.yml
MMathisLab Jun 7, 2024
fecf7e5
Merge branch 'main' into pytorch_docs
MMathisLab Jun 7, 2024
0a2a1c5
Update README.md
MMathisLab Jun 7, 2024
a4e050f
Update installation.md
MMathisLab Jun 7, 2024
9cdb9f1
Delete conda-environments/DEEPLABCUT_M1.yaml
MMathisLab Jun 7, 2024
198a34e
Update DEEPLABCUT.yaml
MMathisLab Jun 7, 2024
e3a5dfa
Update installation.md
MMathisLab Jun 7, 2024
5a774b9
Update maDLC_UserGuide.md
MMathisLab Jun 7, 2024
9d71c9b
Update ModelZoo.md
MMathisLab Jun 8, 2024
3a5a5a1
Update standardDeepLabCut_UserGuide.md
MMathisLab Jun 8, 2024
a7c49eb
Update standardDeepLabCut_UserGuide.md
MMathisLab Jun 8, 2024
0aa9e28
Update standardDeepLabCut_UserGuide.md
MMathisLab Jun 8, 2024
4bf88d2
Update Overviewof3D.md
MMathisLab Jun 8, 2024
665f2df
Update PROJECT_GUI.md
MMathisLab Jun 8, 2024
a388d28
Update UseOverviewGuide.md
MMathisLab Jun 8, 2024
c0028c0
Update beginners-guide.md
MMathisLab Jun 8, 2024
329ce71
Update intro.md
MMathisLab Jun 8, 2024
c1aaa84
Update architectures.md
MMathisLab Jun 8, 2024
16e4386
Update architectures.md
MMathisLab Jun 8, 2024
894e3b4
Merge branch 'main' into pytorch_docs
MMathisLab Jun 8, 2024
680f353
Create course.md
MMathisLab Jun 8, 2024
09bd63e
Update _toc.yml
MMathisLab Jun 8, 2024
d29e7c4
Update manage-project.md
MMathisLab Jun 8, 2024
34a7b5a
Update beginners-guide.md
MMathisLab Jun 8, 2024
20d0bca
Update manage-project.md
MMathisLab Jun 8, 2024
f2336cc
Update _toc.yml
MMathisLab Jun 8, 2024
7a3e0ea
Rename labelling.md to labeling.md
MMathisLab Jun 8, 2024
4802d4f
Update Training-Evaluation.md
MMathisLab Jun 8, 2024
1de7232
Update video-analysis.md
MMathisLab Jun 8, 2024
77ea994
Update architectures.md
MMathisLab Jun 8, 2024
73cd31a
Update architectures.md
MMathisLab Jun 8, 2024
8483175
Update AUTHORS
MMathisLab Jun 8, 2024
518909a
Update architectures.md
MMathisLab Jun 8, 2024
82132e1
doc clean up, tested local build
MMathisLab Jun 8, 2024
0197ba4
Update beginners-guide.md
MMathisLab Jun 8, 2024
e91ee09
Update ModelZoo.md
MMathisLab Jun 8, 2024
daf2f87
Update ModelZoo.md
MMathisLab Jun 8, 2024
2a8949b
Update ModelZoo.md
MMathisLab Jun 8, 2024
9a1f28a
Update ModelZoo.md
MMathisLab Jun 8, 2024
02d71df
Update README.md
MMathisLab Jun 8, 2024
0499825
Update ModelZoo.md
MMathisLab Jun 8, 2024
83db9af
Update ModelZoo.md
MMathisLab Jun 8, 2024
14bbee0
Update AUTHORS
MMathisLab Jun 10, 2024
aa3c82d
Update README.md
MMathisLab Jun 11, 2024
fec76d8
Merge branch 'main' into pytorch_docs
MMathisLab Jun 11, 2024
5927e26
Update installation.md
MMathisLab Jun 12, 2024
3cfb0a8
Update DEEPLABCUT.yaml
MMathisLab Jun 12, 2024
bd212aa
Update DEEPLABCUT.yaml
MMathisLab Jun 12, 2024
a6c9318
Update AUTHORS
MMathisLab Jun 13, 2024
63898d7
fixed typos
Jun 13, 2024
7a33b74
Update python-package.yml
MMathisLab Jun 13, 2024
62ca76b
Update README.md
MMathisLab Jun 14, 2024
7c0a5c1
Update beginners-guide.md
MMathisLab Jun 14, 2024
8dfd1e6
Update user_guide.md
MMathisLab Jun 14, 2024
f6ba9d3
m1 for tf only
MMathisLab Jun 14, 2024
093c424
Merge branch 'main' into pytorch_docs
MMathisLab Jun 14, 2024
aeccccb
Update README.md
MMathisLab Jun 14, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update installation.md
  • Loading branch information
MMathisLab authored Jun 7, 2024
commit e3a5dfaf7a2da79e42a46b20478f0804d4abcc86
51 changes: 28 additions & 23 deletions docs/installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,20 @@

<img src="https://images.squarespace-cdn.com/content/v1/57f6d51c9f74566f55ecf271/71e5d954-75a0-4534-9fa6-7ecc4bf1b76d/installDLC.png?format=1500w" width="250" title="DLC" alt="DLC" align="right" vspace = "50">

#### 🚨 Before you start, do you have a GPU?
````{admonition} 🚨 Click here for more information!
:class: dropdown
- We recommend having a GPU if possible!
- You **need to decide if you want to use a CPU or GPU for your models**: (Note, you can also use the CPU-only for project management and labeling the data! Then, for example, use Google Colaboratory GPUs for free (read more [here](https://github.com/DeepLabCut/DeepLabCut/tree/master/examples#demo-4-deeplabcut-training-and-analysis-on-google-colaboratory-with-googles-gpus) and there are a lot of helper videos on [our YouTube channel!](https://www.youtube.com/playlist?list=PLjpMSEOb9vRFwwgIkLLN1NmJxFprkO_zi)).

- **CPU?** Great, jump to the next section below!

- **NVIDIA GPU?** If you want to use your own GPU (i.e., a GPU is in your workstation), then you need to be sure you have a CUDA compatible GPU, CUDA, and cuDNN installed. Please note, which CUDA you install depends on what version of PyTorch you want to use. So, please check "GPU Support" below carefully. **Note, DeepLabCut is up to date with the latest CUDA and PyTorch!**

- **Apple M-chip GPU?** Be sure to install miniconda3, and your GPU will be used by default.
````


### Step 1: Install Python via Anaconda

#### Install [anaconda](https://www.anaconda.com/distribution/), or use miniconda3 for MacOS users (see below)
Expand Down Expand Up @@ -45,19 +59,12 @@ Windows users: Be sure you have `git` installed along with anaconda: https://git
- click ➡️ for [CONDA FILE](https://github.com/DeepLabCut/DeepLabCut/blob/main/conda-environments/DEEPLABCUT.yaml#:~:text=Raw%20file%20content-,Download,-%E2%8C%98) and then click the "..." and select Download
<img width="274" alt="Screen Shot 2023-09-13 at 10 33 32 PM" src="https://github.com/DeepLabCut/DeepLabCut/assets/28102185/ec4295a5-e85c-4ce7-8c16-e6517a2cfa22">


#### Alternatively, you can git clone this repo and install
- (if the download did not work or you just want to have the source code handy)!

- **Windows/Linux/MacBooks:** git clone this repo (in the terminal/cmd program, while **in a folder** you wish to place DeepLabCut
To git clone type: ``git clone https://github.com/DeepLabCut/DeepLabCut.git``). Note, this can be anywhere, even downloads is fine.)
- **Now, in Terminal (or Anaconda Command Prompt for Windows users)**, if you clicked to download, go to your downloads folder.

```{Hint}
Windows users: Be sure to open the program terminal/cmd/anaconda prompt with a RIGHT-click, "open as admin"
```

- **Now, in Terminal (or Anaconda Command Prompt for Windows users)**, if you clicked to download, go to your downloads folder. Or, if you cloned the repo, go to the DeepLabCut folder.

```{Hint}
:class: dropdown
If you cloned the repo onto your Desktop, the command may look like:
Expand All @@ -66,8 +73,6 @@ You can (on Windows) hold SHIFT and right-click > Copy as path, or (on Mac) righ
```
Be sure you are in the folder that has the `.yaml` file, then run:

- Now, in the terminal run (Windows/Linux/MacBook Intel chip):

``conda env create -f DEEPLABCUT.yaml``


Expand All @@ -80,24 +85,25 @@ NOTE: no need to run pip install deeplabcut, as it is already installed!!! :)

**Great, that's it! DeepLabCut is installed!** 🎉💜

🚨 Next, [head over to the Docs to decide which mode to use DeepLabCut in. You have both standard and multi-animal installed!](https://deeplabcut.github.io/DeepLabCut/docs/UseOverviewGuide.html#what-you-need-to-get-started)
### Step 3: Really, that's it! Let's run DeepLabCut

## Other options:
Head over to the [User Guide Overview](https://deeplabcut.github.io/DeepLabCut/docs/UseOverviewGuide.html) for information.

### PIP:
🎉 Launch DeepLabCut in your new env by running `python -m deeplabcut`

- Everything you need to build custom models within DeepLabCut (i.e., use our source code and our dependencies) can be installed with `pip install 'deeplabcut[gui,tf]'` (for GUI support w/tensorflow) or without the gui: `pip install 'deeplabcut'`.
- If you want to use the SuperAnimal models, then please use `pip install 'deeplabcut[gui,modelzoo]'`.
## Other ways to install DeepLabCut and additional tips

#### We recommend having a GPU.
### Alternatively, you can git clone this repo and install from source!
i.e., if the download did not work or you just want to have the source code handy!

- You **need to decide if you want to use a CPU or GPU for your models**: (Note, you can also use the CPU-only for project management and labeling the data! Then, for example, use Google Colaboratory GPUs for free (read more [here](https://github.com/DeepLabCut/DeepLabCut/tree/master/examples#demo-4-deeplabcut-training-and-analysis-on-google-colaboratory-with-googles-gpus) and there are a lot of helper videos on [our YouTube channel!](https://www.youtube.com/playlist?list=PLjpMSEOb9vRFwwgIkLLN1NmJxFprkO_zi)).
- **Windows/Linux/MacBooks:** git clone this repo (in the terminal/cmd program, while **in a folder** you wish to place DeepLabCut
To git clone type: ``git clone https://github.com/DeepLabCut/DeepLabCut.git``). Note, this can be anywhere, even downloads is fine.)
- Then follow the same steps as in Step 2 above, adjusting for the file now being in the downloaded folder.

- **CPU?** Great, jump to the next section below!
### PIP:

- **NVIDIA GPU?** If you want to use your own GPU (i.e., a GPU is in your workstation), then you need to be sure you have a CUDA compatible GPU, CUDA, and cuDNN installed. Please note, which CUDA you install depends on what version of tensorflow you want to use. So, please check "GPU Support" below carefully. **Note, DeepLabCut is up to date with the latest CUDA and tensorflow versions!**

- **Apple M1/M2 GPU?** Be sure to install miniconda3, and your GPU will be used by default.
- Everything you need to build custom models within DeepLabCut (i.e., use our source code and our dependencies) can be installed with `pip install 'deeplabcut[gui,tf]'` (for GUI support w/tensorflow) or without the gui: `pip install 'deeplabcut'`.
- If you want to use the SuperAnimal models, then please use `pip install 'deeplabcut[gui,modelzoo]'`.

## DOCKER:

Expand All @@ -113,8 +119,7 @@ Here are some conda environment management tips: https://kapeli.com/cheat_sheets

**Pro Tip:** If you want to modify code and then test it, you can use our provided testscripts. This would mean you need to be up-to-date with the latest GitHub-based code though! Please see [here](installTips) on how to get the latest GitHub code, and how to test your installation by following this video: https://www.youtube.com/watch?v=IOWtKn3l33s.


### Creating your own customized conda env (recommended route for Linux: Ubuntu, CentOS, Mint, etc.)
## Creating your own customized conda env (recommended route for Linux: Ubuntu, CentOS, Mint, etc.)

*Note in a fresh ubuntu install, you will often have to run: ``sudo apt-get install gcc python3-dev`` to install the GNU Compiler Collection and the python developing environment.

Expand Down