Regarding GPU and tf compatibility

tensorflow / compression

Data compression in TensorFlow

Apache License 2.0

860 stars 251 forks source link

Regarding GPU and tf compatibility #145

Open apbyju opened 2 years ago

apbyju commented 2 years ago

The GPU that I use to run your code is A100 which is not compatible with CUDA 10. Every time I get the cublass failed routine due to which I am unable to use the HiFiC code. can you suggest me a possible solution on how to use the code for CUDA 11.6 ?

fab-jul commented 2 years ago

Hi, sorry to hear that. I assume this is because we rely on TF1 and I do not know if there is away to upgrade CUDA for TF1 :(