neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-21 04:34:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #760 from Fmstrat/docker-fix

Browse source 
Update Dockerfile to enable CUDA pytorch
This commit is contained in:
manmay nakhashi 2024-05-09 18:03:35 +05:30 committed by GitHub
parent eb4c4668ab 61d88a6166
commit 572bdf3d24
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Dockerfile
View file

@ -28,7 +28,7 @@ SHELL ["/bin/bash", "--login", "-c"]
RUN conda create --name tortoise python=3.9 numba inflect \
&& conda activate tortoise \
&& conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia \
&& conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
&& conda install transformers=4.31.0 \
&& cd /app \
&& python setup.py install