Update README.md

This commit is contained in:
天涯古巷
2024-01-07 07:07:58 +08:00
committed by GitHub
parent 2d1eb11fd6
commit 4abc3409ac

View File

@@ -138,11 +138,11 @@ conda activate videoEnv
- 安装GPU版本Pytorch: - 安装GPU版本Pytorch:
```shell ```shell
conda install pytorch==2.1.0 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia conda install pytorch==2.0.1 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia
``` ```
或者使用 或者使用
```shell ```shell
pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118 pip install torch==2.0.1 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
``` ```
- 安装其他依赖: - 安装其他依赖: