gpu10: pytorch and cuda

[Yichong Xu]

It seems like tensorflow does not support cuda10 right now - it has to be installed from source.
But I’m mainly using pytorch though and the version with cuda10 does not run either.
Plus, I tried the original cuda example and it cannot find the gpu either:
(base) yichongx at gpu10$ ls
Makefile  readme.txt  simplePrintf  simplePrintf.cu  simplePrintf.o
(base) yichongx at gpu10$ ./simplePrintf
CUDA error at ../../common/inc/helper_cuda.h:744 code=999(cudaErrorUnknown) "cudaGetDeviceCount(&device_count)"
(base) yichongx at gpu10$



Thanks,
Yichong



On Mar 10, 2019, at 9:52 AM, Yotam Hechtlinger <yhechtli at andrew.cmu.edu<mailto:yhechtli at andrew.cmu.edu>> wrote:

It's not the same cuda version on GPU 10 and the rest, I think different version of tensorflow has to be installed.

Check your tensorflow version and if it supports the cuda version on gpu10.



On Saturday, March 9, 2019, Predrag Punosevac <predragp at andrew.cmu.edu<mailto:predragp at andrew.cmu.edu>> wrote:
Try CUDA 10.0 instead of 10.1

On Mar 9, 2019 5:28 PM, Yichong Xu <yichongx at cs.cmu.edu<mailto:yichongx at cs.cmu.edu>> wrote:
Same issue here.

From my iPhone

On Mar 9, 2019, at 4:01 PM, Emre Yolcu <eyolcu at andrew.cmu.edu<mailto:eyolcu at andrew.cmu.edu>> wrote:


Hi,



Right now on gpu10 `nvcc --version` and `nvidia-smi` seem to work, but `python -c ‘import torch; print(torch.cuda.is_available())’` prints False. Is anybody running into the same issue?



Emre


-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mailman.srv.cs.cmu.edu/pipermail/autonlab-users/attachments/20190310/ac5ca2d5/attachment.html>