Is 8G GPU enough for pytorch training?

Task (what are you trying to do/register?)

[please describe task here]

What have you tried

Please describe specifics of your approach // use of vxm

Details of experiments

Please carefully specify details about your experiments. If you are training, when what is the setup? What loss are you using? What does the convergence look like? If you are registering, please show example inputs and outputs. etc.

[klfxmu]

You’re the most patient writer I’ve ever met。I’ll go over the information you gave me right away Thank you

On 02/24/2021 21:42, Adrian Dalca wrote:

@klfxmu the instructions we have in the readme won’t determine if your code uses cpu or gpu, that’s up to the platform you use (e.g. keras) and the installation you have. For most people with tensorflow-gpu installed, those instructions will run voxelmorph on the gpu.

This probably has to do with your tensorflow/keras installation. So I would recommend reading into this. e.g.:

https://stackoverflow.com/questions/44829085/tensorflow-not-running-on-gpu

https://stackoverflow.com/questions/64467035/tensorflow-uses-cpu-instead-of-gpu

https://stackoverflow.com/questions/63016659/why-tensorflow-not-running-on-gpu-while-gpu-devices-are-identified-in-python

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or unsubscribe.

[adalca]

@klfxmu the instructions we have in the readme won’t determine if your code uses cpu or gpu, that’s up to the platform you use (e.g. keras) and the installation you have. For most people with tensorflow-gpu installed, those instructions will run voxelmorph on the gpu.

This probably has to do with your tensorflow/keras installation. So I would recommend reading into this. e.g.:

https://stackoverflow.com/questions/44829085/tensorflow-not-running-on-gpu

https://stackoverflow.com/questions/64467035/tensorflow-uses-cpu-instead-of-gpu

https://stackoverflow.com/questions/63016659/why-tensorflow-not-running-on-gpu-while-gpu-devices-are-identified-in-python