yajiedesign commented on a change in pull request #9798: fix cmake
URL: https://github.com/apache/incubator-mxnet/pull/9798#discussion_r169942811
 
 

 ##########
 File path: CMakeLists.txt
 ##########
 @@ -46,6 +46,11 @@ if(USE_CUDA AND NOT USE_OLDCMAKECUDA)
         (${CMAKE_VERSION} VERSION_GREATER "3.9.0") OR (${CMAKE_VERSION} 
VERSION_EQUAL "3.9.0")
       )
     )
+    
+    set(__cuda_toolset "7.5" "8.0" "9.0")
+    set(CUDA_TOOLSET "8.0" CACHE STRING "Select CUDA Version.")
 
 Review comment:
   sorry for later,if not set CUDA_TOOLSET,it will use the latest version cuda.
   If you install multiple cuda versions at the same time, you need 
CUDA_TOOLSET to select the version.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


With regards,
Apache Git Services

Reply via email to