TFDeepSurv icon indicating copy to clipboard operation
TFDeepSurv copied to clipboard

Published 20 hours ago • verified publisher icon liupei101

gpu support

Open bnuzyc91 opened this issue 4 years ago • 2 comments

@liupei101 Thank you so much for the package.

Could you please advise is it possible to use GPU support for this TFDeepSurv package?

Thank you!

bnuzyc91 avatar Feb 07 '21 10:02 bnuzyc91

Only FC layers are used in DeepSurv model, not involving Conv etc, but you can modify code to adapt your environment (GPU or CPU).

PRs are welcomed!

liupei101 avatar Mar 05 '21 01:03 liupei101

I tried to find a way to use the GPU without modifying the code and since I notice that you have utilized the tensorflow 1.* version for TFDeepSurv. I have done the following:

  1. install the TensorFlow 1.15-GPU version using pip3 install tensorflow-gpu==1.15
  2. run the TFDeepSurv code under TensorFlow 1.15-GPU using my GPU[0] I notice that the percentage of GPU utilization is in the range of 0%~3% [0] TITAN RTX | 60'C, 3 % | 23216 / 24217 MB |

I am not sure if this actually uses the GPU or not. I think it uses since I can see the temperature of GPU raise and the percentage is not always 0%. But on the other hand, I am not sure why the GPU utilization is low.

Could you please advise if I did the right thing? Thank you!

bnuzyc91 avatar Mar 08 '21 03:03 bnuzyc91