Docker如何启用GPU加速 about vits-simple-api HOT 6 CLOSED

Dockerfile是构建docker镜像时使用的，所以这里更改Dockerfile不会起作用。

之前我觉得使用docker的大部分都是部署在没有gpu的小型服务器上，所以在某次更新中设置为下载cpu版pytorch以减小docker镜像的体积。晚点可以添加一个安装gpu版pytorch的docker镜像。

from vits-simple-api.

Dockerfile是构建docker镜像时使用的，所以这里更改Dockerfile不会起作用。

之前我觉得使用docker的大部分都是部署在没有gpu的小型服务器上，所以在某次更新中设置为下载cpu版pytorch以减小docker镜像的体积。晚点可以添加一个安装gpu版pytorch的docker镜像。

我使用的是自己构建的镜像。

from vits-simple-api.

那在Dockerfile中安装pytorch的命令改成RUN pip install torch --index-url https://download.pytorch.org/whl/cu117然后重新构建试试，最后构建出的镜像应该是5GB左右

from vits-simple-api.

那在Dockerfile中安装pytorch的命令改成RUN pip install torch --index-url https://download.pytorch.org/whl/cu117然后重新构建试试，最后构建出的镜像应该是5GB左右

我修改了dockerfile并重新构建了，但是依旧不可用，我实在不知道出现了什么问题。
基本相同的配置在我的一个仿照moe-tts的项目中是可行的，你可以在这里看到它的代码https://huggingface.co/spaces/RyaoChengfeng/vits-moe

不过我也没有在gpu服务器上跑它的需求:)

from vits-simple-api.

我也尝试了改成RUN pip install torch --index-url https://download.pytorch.org/whl/cu117后，docker安装的pytorch版本确实变为了torch:2.0.1+cu117，再配置好gpu应该可以用才对。然而我也没有gpu服务器去测试和验证 😢

from vits-simple-api.

我也尝试了改成RUN pip install torch --index-url https://download.pytorch.org/whl/cu117后，docker安装的pytorch版本确实变为了torch:2.0.1+cu117，再配置好gpu应该可以用才对。然而我也没有gpu服务器去测试和验证 😢

此处当有广告位，赶紧来家云厂商或者GPU厂商赞助一台GPU机器

from vits-simple-api.