From c2e71c93687c0d57e012fbd2024b8a46f4e37b11 Mon Sep 17 00:00:00 2001
From: Peter Baylies <pbaylies@gmail.com>
Date: Mon, 21 Dec 2020 01:56:20 -0500
Subject: [PATCH] Update install script for runway
---
runway.yml | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/runway.yml b/runway.yml
index e026930..a350de3 100644
--- a/runway.yml
+++ b/runway.yml
@@ -1,7 +1,7 @@
version: 0.1
entrypoint: python runway_model.py
python: 3.6
-cuda: 10.0
+cuda: 10.2
framework: tensorflow
spec:
gpu: True
@@ -10,5 +10,7 @@ files:
ignore:
- checkpoints/*
build_steps:
- - apt-get install -y cuda-nvcc-10-0 cuda-toolkit-10-0
+ - apt-get update
+ - apt-get upgrade -y
+ - apt-get install -y cuda-nvcc-10-2 cuda-toolkit-10-2
- pip install -r requirements.txt
--
GitLab