Merge pull request #232 from kknox/update-cuda-version

Update container to newer cuda driver and sdk 9.0

[ROCm/clr commit: cf9eb832a2]
Этот коммит содержится в:
Ben Sander
2017-10-25 23:45:24 +02:00
коммит произвёл GitHub
родитель 5a2bdb98f0 f32e20abc7
Коммит 0a82572fa2
поставляемый
+4 -4
Просмотреть файл
@@ -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 )