mxnet-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-mxnet] piyushghai commented on issue #15254: mxnet(mxnet-full_2.11-linux-x86_64-gpu-1.5.0-SNAPSHOT) cannot support cuda10.1?
Date Tue, 18 Jun 2019 17:37:38 GMT
piyushghai commented on issue #15254: mxnet(mxnet-full_2.11-linux-x86_64-gpu-1.5.0-SNAPSHOT)
cannot support cuda10.1?
URL: https://github.com/apache/incubator-mxnet/issues/15254#issuecomment-503237205
 
 
   > **Thank you all. I lowered the CUDA version(9.2), and now it's OK. But my code can't
use GPU. Do you have any suggestions?**
   > 
   > ```java
   > @Test
   > public void testGPU(){
   >      System.out.println(System.getenv().containsKey("SCALA_TEST_ON_GPU"));
   > }
   > 
   >  //  result is false.
   > ```
   
   @tomoncle This is merely an environment variable that is used for test/debugging purposes.

   
   In order to make the package use GPU, you need to set the context to use GPU. 
   ```List<Context> context = new ArrayList<Context>();
         context.add(Context.gpu());
   ```

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


With regards,
Apache Git Services

Mime
View raw message