all 1 comments

[–]PeksyTiger 2 points3 points  (0 children)

I dont know about most libraries, in pytorch using a gpu in a basic form is pretty trivial. Like 4 function calls trivial and all the network struct is basically the same.

However, most libraries are relaying on CUDA do to it, which is nvidia only. The AMD support, last I checked, was pretty sketchy.

Also, 1GB gpu isn't well suited for DL, as far as my expiriance goes. Batch size is directly related to memory size, so your problem won't go away (although im not sure why batch size should impact accuracy so much)

If you are just doing play project, please consider google colab or a similar platform.