Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
JunjieLl committed Oct 28, 2024
1 parent aff1163 commit 470f635
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 8 deletions.
8 changes: 8 additions & 0 deletions .github/workflows/env_linux.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,14 @@ jobs:
build:
runs-on: ubuntu-latest
steps:
# - name: free disk space
# run: |
# sudo swapoff -a
# sudo rm -f /swapfile
# sudo apt clean
# docker rmi $(docker image ls -aq)
# df -h

- name: Login to Docker Hub
uses: docker/login-action@v3
with:
Expand Down
10 changes: 5 additions & 5 deletions environment/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
FROM nvidia/cuda:11.6.1-cudnn8-runtime-ubuntu20.04
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime

ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Asia/Shanghai
# ENV DEBIAN_FRONTEND=noninteractive
# ENV TZ=Asia/Shanghai

COPY ./requirements.txt /tmp/requirements.txt

RUN apt-get update && \
apt-get upgrade -y && \
apt-get install -y sudo build-essential python3.10 python3-pip vim wget git libsndfile1 espeak-ng sox zip && \
apt-get install -y sudo build-essential vim wget git libsndfile1 espeak-ng sox zip && \
apt-get clean && \
pip3 install -r /tmp/requirements.txt --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu116 && \
pip install -r /tmp/requirements.txt --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu116 && \
rm /tmp/requirements.txt
4 changes: 1 addition & 3 deletions environment/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
torch==1.13.1+cu116
torchvision==0.14.1+cu116
torchaudio==0.13.1
torchaudio==0.13.1
auraloss==0.4.0
cython==3.0.10
einops
Expand Down

0 comments on commit 470f635

Please sign in to comment.