4 ms·
For Pytorch and Tensorflow, you can use conda to install them with the right CUDA and cudnn versions.
by wdroz 5y ago
For Pytorch and Tensorflow, you can use conda to install them with the right CUDA and cudnn versions.
- kwertzzz 5y agoFor 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.
- wdroz 5y agoYou install cudatoolkit from conda then tensorflow with pip.