diff --git a/runway.yml b/runway.yml index 3353e754cda2f5949c91fe17a000d491f0d3e445..a07bf08ab199fd5965bbb6d270bf7afa431e662f 100644 --- a/runway.yml +++ b/runway.yml @@ -1,7 +1,7 @@ version: 0.1 entrypoint: python runway_model.py python: 3.6 -cuda: 10.2 +cuda: 10.0 framework: tensorflow spec: gpu: True @@ -12,5 +12,5 @@ files: build_steps: - apt-get update > install-update.log 2>&1 - apt-get upgrade -y > install-upgrade.log 2>&1 - - apt-get install -y cuda-nvcc-10-2 cuda-toolkit-10-2 > install-package.log 2>&1 + - apt-get install -y cuda-nvcc-10-0 cuda-toolkit-10-0 > install-package.log 2>&1 - pip install -r requirements.txt