Unverified Commit fe921bfa authored by moto's avatar moto Committed by GitHub
Browse files

Update GPU test image for development (#1191)

Changing the test Docker image for GPU tests to use Nvidia Docker `devel` image. This is necessary for #1187. 
parent c257ee7d
...@@ -430,7 +430,7 @@ jobs: ...@@ -430,7 +430,7 @@ jobs:
resource_class: gpu.small resource_class: gpu.small
environment: environment:
<<: *environment <<: *environment
image_name: "pytorch/torchaudio_unittest_base:manylinux-cuda10.1" image_name: pytorch/torchaudio_unittest_base:manylinux-cuda10.1-cudnn7-20201203
steps: steps:
- checkout - checkout
- attach_workspace: - attach_workspace:
......
...@@ -430,7 +430,7 @@ jobs: ...@@ -430,7 +430,7 @@ jobs:
resource_class: gpu.small resource_class: gpu.small
environment: environment:
<<: *environment <<: *environment
image_name: "pytorch/torchaudio_unittest_base:manylinux-cuda10.1" image_name: pytorch/torchaudio_unittest_base:manylinux-cuda10.1-cudnn7-20201203
steps: steps:
- checkout - checkout
- attach_workspace: - attach_workspace:
......
...@@ -7,15 +7,14 @@ if [ $# -ne 1 ]; then ...@@ -7,15 +7,14 @@ if [ $# -ne 1 ]; then
exit 1 exit 1
fi fi
datestr="$(date "+%Y%m%d")"
if [ "$1" = "cpu" ]; then if [ "$1" = "cpu" ]; then
base_image="ubuntu:18.04" base_image="ubuntu:18.04"
image="pytorch/torchaudio_unittest_base:manylinux" image="pytorch/torchaudio_unittest_base:manylinux-${datestr}"
elif [[ "$1" =~ ^(9.2|10.1)$ ]]; then
base_image="nvidia/cuda:$1-runtime-ubuntu18.04"
image="pytorch/torchaudio_unittest_base:manylinux-cuda$1"
else else
printf "Unexpected <CUDA_VERSION> string: %s" "$1" base_image="nvidia/cuda:$1-devel-ubuntu18.04"
exit 1; docker pull "${base_image}"
image="pytorch/torchaudio_unittest_base:manylinux-cuda$1-${datestr}"
fi fi
cd "$( dirname "${BASH_SOURCE[0]}" )" cd "$( dirname "${BASH_SOURCE[0]}" )"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment