Open ytzhao opened 6 years ago
I am having this exact issue. Is there any fix you have found for this yet?
@stancil1 Hi, I change the CUDA version to 9.0, it somehow works.
I'm finding that the torch.cudnn package is not properly initializing the Volta GPUs (it takes about 10 minutes to configure the GPUs -- it should be nearly instantaneous). Were you able to configure GPUs properly using cutorch?
This is the code that cudnn is having trouble with:
Hi all, I tried to install torch7 on Tesla V100, but met some problems. For some reasons, I have to use CUDA8.0(although for Volta architecture, it's better to use CUDA9.0).
I think the problem is the gpu architecture dismatch cuda version. (for Volta architecture, it officially needs CUDA9.0, it also works with CUDA8.0). How could I disable the auto-detection, and manually set it to other runnable version, such as "compute_60" or somehow? Thank you.