[CI] Update github workflow cuda image (#874)

* update workflow

* another trial

* another trial

* another trial
This commit is contained in:
Fengzhe Zhou 2024-02-05 15:22:59 +08:00 committed by GitHub
parent fed7d800c6
commit fc84aff963
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -53,7 +53,7 @@ jobs:
build_cu117:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
image: nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04
strategy:
matrix:
python-version: ['3.10']