Open dongheechoi opened 3 months ago
I tried to run the script but I don't think it is using a GPU. I stopped the code before this line (https://github.com/ncbi/BioREx/blob/0031c52d2dd0d7fb8e3b84e256d8fbf73c4bd463/src/tf_wrapper.py#L199C28-L200C1), and check it whether using GPU or not, but I got this.
In [16]: self.model.layers[0].variables[0].device Out[16]: '/job:localhost/replica:0/task:0/device:CPU:0'
Can you kindly check the script or runfile to use GPU to train?
And I think it is better to change the static gpu variable in the script into dynamic one as you put in the other script. I also put that in the pull request(https://github.com/ncbi/BioREx/pull/4) https://github.com/ncbi/BioREx/blob/0031c52d2dd0d7fb8e3b84e256d8fbf73c4bd463/scripts/run_biorex_exp.sh#L3
Hi @dongheechoi ,
The program can run on GPU as below.
Yes, you should be able to change cuda_visible_devices to the visible GPUs.
I tried to run the script but I don't think it is using a GPU. I stopped the code before this line (https://github.com/ncbi/BioREx/blob/0031c52d2dd0d7fb8e3b84e256d8fbf73c4bd463/src/tf_wrapper.py#L199C28-L200C1), and check it whether using GPU or not, but I got this.
Can you kindly check the script or runfile to use GPU to train?
And I think it is better to change the static gpu variable in the script into dynamic one as you put in the other script. I also put that in the pull request(https://github.com/ncbi/BioREx/pull/4) https://github.com/ncbi/BioREx/blob/0031c52d2dd0d7fb8e3b84e256d8fbf73c4bd463/scripts/run_biorex_exp.sh#L3