cuda fix

cantabile-kwok · Oct 28, 2024 · 6ce2cb9 · 6ce2cb9
1 parent d013dac
commit 6ce2cb9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/environment/Dockerfile b/environment/Dockerfile
@@ -1,7 +1,4 @@
-FROM python:3.10.8
-
-ENV TZ=Asia/Shanghai
-ENV DEBIAN_FRONTEND=noninteractive
+FROM FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
 
 COPY ./requirements.txt /tmp/requirements.txt
 

diff --git a/environment/requirements.txt b/environment/requirements.txt
@@ -1,5 +1,3 @@
-torch==1.13.1+cu116
-torchvision==0.14.1+cu116
 torchaudio==0.13.1 
 auraloss==0.4.0
 cython==3.0.10