235
235
< div class ="pytorch-left-menu-search ">
236
236
237
237
< div class ="version ">
238
- < a href ='https://pytorch.org/docs/versions.html '> master (2.1.0a0+gitf3cf3d7 ) ▼</ a >
238
+ < a href ='https://pytorch.org/docs/versions.html '> master (2.1.0a0+git129e039 ) ▼</ a >
239
239
</ div >
240
240
241
241
@@ -538,6 +538,12 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
538
538
539
539
< span class ="n "> dll_paths</ span > < span class ="o "> =</ span > < span class ="nb "> list</ span > < span class ="p "> (</ span > < span class ="nb "> filter</ span > < span class ="p "> (</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> exists</ span > < span class ="p "> ,</ span > < span class ="p "> [</ span > < span class ="n "> th_dll_path</ span > < span class ="p "> ,</ span > < span class ="n "> py_dll_path</ span > < span class ="p "> ,</ span > < span class ="n "> base_py_dll_path</ span > < span class ="p "> ]))</ span >
540
540
541
+ < span class ="k "> if</ span > < span class ="nb "> all</ span > < span class ="p "> ([</ span > < span class ="ow "> not</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> exists</ span > < span class ="p "> (</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> join</ span > < span class ="p "> (</ span > < span class ="n "> p</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'nvToolsExt64_1.dll'</ span > < span class ="p "> ))</ span > < span class ="k "> for</ span > < span class ="n "> p</ span > < span class ="ow "> in</ span > < span class ="n "> dll_paths</ span > < span class ="p "> ]):</ span >
542
+ < span class ="n "> nvtoolsext_dll_path</ span > < span class ="o "> =</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> join</ span > < span class ="p "> (</ span >
543
+ < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> getenv</ span > < span class ="p "> (</ span > < span class ="s1 "> 'NVTOOLSEXT_PATH'</ span > < span class ="p "> ,</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> join</ span > < span class ="p "> (</ span > < span class ="n "> pfiles_path</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'NVIDIA Corporation'</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'NvToolsExt'</ span > < span class ="p "> )),</ span > < span class ="s1 "> 'bin'</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'x64'</ span > < span class ="p "> )</ span >
544
+ < span class ="k "> else</ span > < span class ="p "> :</ span >
545
+ < span class ="n "> nvtoolsext_dll_path</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
546
+
541
547
< span class ="kn "> from</ span > < span class ="nn "> .version</ span > < span class ="kn "> import</ span > < span class ="n "> cuda</ span > < span class ="k "> as</ span > < span class ="n "> cuda_version</ span >
542
548
< span class ="kn "> import</ span > < span class ="nn "> glob</ span >
543
549
< span class ="k "> if</ span > < span class ="n "> cuda_version</ span > < span class ="ow "> and</ span > < span class ="nb "> all</ span > < span class ="p "> ([</ span > < span class ="ow "> not</ span > < span class ="n "> glob</ span > < span class ="o "> .</ span > < span class ="n "> glob</ span > < span class ="p "> (</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> join</ span > < span class ="p "> (</ span > < span class ="n "> p</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'cudart64*.dll'</ span > < span class ="p "> ))</ span > < span class ="k "> for</ span > < span class ="n "> p</ span > < span class ="ow "> in</ span > < span class ="n "> dll_paths</ span > < span class ="p "> ]):</ span >
@@ -548,7 +554,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
548
554
< span class ="k "> else</ span > < span class ="p "> :</ span >
549
555
< span class ="n "> cuda_path</ span > < span class ="o "> =</ span > < span class ="s1 "> ''</ span >
550
556
551
- < span class ="n "> dll_paths</ span > < span class ="o "> .</ span > < span class ="n "> extend</ span > < span class ="p "> (</ span > < span class ="nb "> filter</ span > < span class ="p "> (</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> exists</ span > < span class ="p "> ,</ span > < span class ="p "> [</ span > < span class ="n "> cuda_path</ span > < span class ="p "> ]))</ span >
557
+ < span class ="n "> dll_paths</ span > < span class ="o "> .</ span > < span class ="n "> extend</ span > < span class ="p "> (</ span > < span class ="nb "> filter</ span > < span class ="p "> (</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> path</ span > < span class ="o "> .</ span > < span class ="n "> exists</ span > < span class ="p "> ,</ span > < span class ="p "> [</ span > < span class ="n "> nvtoolsext_dll_path </ span > < span class =" p " > , </ span > < span class =" n " > cuda_path</ span > < span class ="p "> ]))</ span >
552
558
553
559
< span class ="n "> kernel32</ span > < span class ="o "> =</ span > < span class ="n "> ctypes</ span > < span class ="o "> .</ span > < span class ="n "> WinDLL</ span > < span class ="p "> (</ span > < span class ="s1 "> 'kernel32.dll'</ span > < span class ="p "> ,</ span > < span class ="n "> use_last_error</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> )</ span >
554
560
< span class ="n "> with_load_library_flags</ span > < span class ="o "> =</ span > < span class ="nb "> hasattr</ span > < span class ="p "> (</ span > < span class ="n "> kernel32</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'AddDllDirectory'</ span > < span class ="p "> )</ span >
0 commit comments