[queuebert]

I just finished setting up a new machine to run some Kaggle stuff. Both Tensorflow and PyTorch had issues with CUDA versions and dependencies that weren't immediately fixed with a clean virtualenv, while both Knet.jl and Flux.jl installed flawlessly.

[wdroz]

For Pytorch and Tensorflow, you can use conda to install them with the right CUDA and cudnn versions.

[kwertzzz]

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.

[wdroz]

You install cudatoolkit from conda then tensorflow with pip.