mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-02-04 04:34:41 +08:00
Update README.md
This commit is contained in:
12
README.md
12
README.md
@@ -137,12 +137,12 @@ conda activate videoEnv
|
||||
|
||||
- 安装GPU版本Pytorch:
|
||||
|
||||
```shell
|
||||
conda install pytorch==2.0.1 torchvision==0.15.2 pytorch-cuda=11.7 -c pytorch -c nvidia
|
||||
```shell
|
||||
conda install pytorch==2.1.0 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia
|
||||
```
|
||||
或者使用
|
||||
```shell
|
||||
pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
|
||||
```shell
|
||||
pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
|
||||
```
|
||||
|
||||
- 安装其他依赖:
|
||||
@@ -214,6 +214,10 @@ LAMA_SUPER_FAST = False # 保证效果
|
||||
|
||||
解决方案:改用cuda 11.8
|
||||
|
||||
```shell
|
||||
pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
|
||||
```
|
||||
|
||||
## 赞助
|
||||
<img src="https://i.imgur.com/EMCP5Lv.jpeg" width="600">
|
||||
|
||||
|
||||
Reference in New Issue
Block a user