Update container to newer cuda driver and sdk 9.0

Tá an tiomantas seo le fáil i:
Kent Knox
2017-10-25 16:14:32 -05:00
tuismitheoir 4e78738267
tiomantas f81d6d5bb3
+4 -4
Féach ar an gComhad
@@ -447,13 +447,13 @@ nvcc:
{
////////////////////////////////////////////////////////////////////////
// Block of string constants customizing behavior for cuda
String nvcc_ver = 'nvcc-8.0'
String from_image = 'nvidia/cuda:8.0-devel'
String nvcc_ver = 'nvcc-9.0'
String from_image = 'nvidia/cuda:9.0-devel'
// This unfortunately hardcodes the driver version nvidia_driver_375.74 in the volume mount. Research if a way
// This unfortunately hardcodes the driver version nvidia_driver_384.90 in the volume mount. Research if a way
// exists to get volume driver to customize the volume names to leave out driver version
String inside_args = '''--device=/dev/nvidiactl --device=/dev/nvidia0 --device=/dev/nvidia-uvm --device=/dev/nvidia-uvm-tools
--volume-driver=nvidia-docker --volume=nvidia_driver_375.74:/usr/local/nvidia:ro''';
--volume-driver=nvidia-docker --volume=nvidia_driver_384.90:/usr/local/nvidia:ro''';
// Checkout source code, dependencies and version files
String source_hip_rel = checkout_and_version( nvcc_ver )