fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
David PAGNON 2024-09-03 11:10:59 +02:00 committed by GitHub
parent 14bf609aec
commit d925177ce2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -135,12 +135,6 @@ If you don't use Anaconda, type `python -V` in terminal to make sure python>=3.9
``` cmd ``` cmd
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
``` ```
***Note:*** issues reported with the default command. However, this has been tested and works:
`pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu118`
Then install ONNX Runtime with GPU support: Then install ONNX Runtime with GPU support:
``` ```
pip install onnxruntime-gpu pip install onnxruntime-gpu