Fix docker image to run on my system's host CUDA version

2026-10-29 09:18:21 +00:00 · 2022-12-12 20:54:11 -05:00
parent 3f0f94d92e
commit b1dca3ba63
3 changed files with 23 additions and 9 deletions
@@ -42,9 +42,6 @@ soundfile==0.11.0
 soupsieve==2.3.2.post1
 stack-data==0.6.2
 threadpoolctl==3.1.0
-torch==1.13.0+cu116
-torchaudio==0.13.0+cu116
-torchvision==0.14.0+cu116
 tqdm==4.64.1
 traitlets==5.6.0
 typing_extensions==4.4.0