File tree 1 file changed +3
-0
lines changed
1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -114,6 +114,9 @@ private static void LoadNativeBackend(bool useCudaBackend, out StringBuilder? tr
114
114
ok = TryLoadNativeLibraryByName ( "cudnn_adv64_9" , typeof ( torch ) . Assembly , trace ) ;
115
115
ok = TryLoadNativeLibraryByName ( "cudnn_cnn64_9" , typeof ( torch ) . Assembly , trace ) ;
116
116
ok = TryLoadNativeLibraryByName ( "cudnn_ops64_9" , typeof ( torch ) . Assembly , trace ) ;
117
+ ok = TryLoadNativeLibraryByName ( "cudnn_graph64_9.dll" , typeof ( torch ) . Assembly , trace ) ;
118
+ ok = TryLoadNativeLibraryByName ( "cudnn_engines_precompiled64_9.dll" , typeof ( torch ) . Assembly , trace ) ;
119
+ ok = TryLoadNativeLibraryByName ( "cudnn_engines_runtime_compiled64_9.dll" , typeof ( torch ) . Assembly , trace ) ;
117
120
ok = TryLoadNativeLibraryByName ( "nvrtc-builtins64_121" , typeof ( torch ) . Assembly , trace ) ;
118
121
ok = TryLoadNativeLibraryByName ( "caffe2_nvrtc" , typeof ( torch ) . Assembly , trace ) ;
119
122
ok = TryLoadNativeLibraryByName ( "nvrtc64_120_0" , typeof ( torch ) . Assembly , trace ) ;
You can’t perform that action at this time.
0 commit comments