kromskii2/alltalk_tts
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #609 from Paladinium/fix_pytorch_version

Browse Source 
Fix pytorch version to 2.6.0
This commit is contained in:
Paladinium authored and GitHub committed 2025-06-09 13:51:20 +02:00
2 parents 38683ae9c8 d33dde7fc8
commit ee4477e221
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
docker/base/Dockerfile
+1 -1
View File
@@ -76,7 +76,7 @@ RUN <<EOR
conda activate alltalk conda activate alltalk
mkdir -p ${ALLTALK_DIR}/pip_cache mkdir -p ${ALLTALK_DIR}/pip_cache
pip install --no-cache-dir --cache-dir=${ALLTALK_DIR}/pip_cache \ pip install --no-cache-dir --cache-dir=${ALLTALK_DIR}/pip_cache \
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu${CUDA_SHORT_VERSION_NO_DOT} torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_SHORT_VERSION_NO_DOT}
if [ $? -ne 0 ] ; then if [ $? -ne 0 ] ; then
echo "Failed to install pip dependencies: $RESULT" echo "Failed to install pip dependencies: $RESULT"