Re:Cuda.jl v3.3: union types, debug info, graph APIs
by wdroz on Monday June 14, 2021 @02:44AM
(#27499944)
For Pytorch and Tensorflow, you can use conda to install them with the right CUDA and cudnn versions.
[ Parent
| on HN ]
Re:Cuda.jl v3.3: union types, debug info, graph APIs
by kwertzzz on Monday June 14, 2021 @03:18AM
(#27500155)
For Pytorch, I had no issues with conda. But with Tensorflow from conda, the training process just hangs (consuming 100% of CPU but no GPU resources, despite my GPUs are recognized). I got more luck with installing Tensorflow with pip. Given the fact that Tensorflow documentation does not mention conda, I wondering how well this is supported.