Create CPU image
Rexhaif opened this issue · 1 comments
Rexhaif commented
While CUDA images can work on cpu, they are quite heavy. So it's reasonable to create cpu-only images, without heavy CUDA libs. Also, for cpu-only version we can use cpu-only Pytorch and Tensorflow, which are significantly lighter that default gpu versions.
Rexhaif commented
Done, created :cpu tag