Skip to content

Commit e7149de

Browse files
authored
add snapshot of CUDA deps
1 parent 6d70ecd commit e7149de

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

test/cuda/layers.jl

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,20 @@ function gpu_gradtest(name::String, layers::Vector, x_cpu = nothing, args...; te
4646
# test
4747
if test_cpu
4848
if VERSION >= v"1.7" && layer === GroupedConvTranspose && args[end] == selu
49-
# FIXME revisit this after CUDA deps on CI are updated
49+
# FIXME revisit this after CUDA deps on CI are updated. Here's a snapshot from 2021-12-25:
50+
# CUDA toolkit 11.4, local installation
51+
# NVIDIA driver 495.29.5, for CUDA 11.5
52+
# CUDA driver 11.5
53+
# Libraries:
54+
# - CUBLAS: 11.6.1
55+
# - CURAND: 10.2.5
56+
# - CUFFT: 10.5.2
57+
# - CUSOLVER: 11.2.0
58+
# - CUSPARSE: 11.6.0
59+
# - CUPTI: 14.0.0
60+
# - NVML: 11.0.0+495.29.5
61+
# - CUDNN: 8.20.4 (for CUDA 11.4.0)
62+
# - CUTENSOR: missing
5063
@test y_gpu y_cpu rtol=2 atol=2
5164
else
5265
@test y_gpu y_cpu rtol=1f-3 atol=1f-3

0 commit comments

Comments
 (0)