use torch from docker image

pull/2187/head
RunningLeon 2023-06-26 13:11:59 +08:00
parent a8fe7aa1b1
commit fdb48404fc
1 changed files with 2 additions and 25 deletions
.github/workflows

View File

@ -149,13 +149,6 @@ jobs:
image: pytorch/pytorch:1.9.0-cuda10.2-cudnn7-devel
env:
FORCE_CUDA: 1
strategy:
matrix:
torch: [1.9.0+cu102]
include:
- torch: 1.9.0+cu102
torch_version: torch1.9
torchvision: 0.10.0+cu102
steps:
- uses: actions/checkout@v2
- name: Install system dependencies
@ -164,14 +157,10 @@ jobs:
apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libxrender-dev
apt-get clean
rm -rf /var/lib/apt/lists/*
- name: Install PyTorch
run: |
python -V
python -m pip install --upgrade pip
python -m pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
- name: Install dependencies
run: |
python -V
python -m pip show torch torchvision
export CFLAGS=`python -c 'import sysconfig;print("-I"+sysconfig.get_paths()["include"])'`
python -m pip install openmim
python -m pip install -r requirements.txt
@ -204,13 +193,6 @@ jobs:
runs-on: ubuntu-20.04
container:
image: pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
strategy:
matrix:
torch: [1.12.0+cu113]
include:
- torch: 1.12.0+cu113
torch_version: torch1.12
torchvision: 0.13.0+cu113
steps:
- uses: actions/checkout@v2
- name: Install system dependencies
@ -219,15 +201,10 @@ jobs:
apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libxrender-dev
apt-get clean
rm -rf /var/lib/apt/lists/*
- name: Install PyTorch
run: |
python -V
python -m pip list
python -m pip install --upgrade pip
python -m pip install --no-cache-dir torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
- name: Install dependencies
run: |
python -V
python -m pip show torch torchvision
export CFLAGS=`python -c 'import sysconfig;print("-I"+sysconfig.get_paths()["include"])'`
python -m pip install --no-cache-dir openmim
python -m pip install --no-cache-dir -r requirements.txt