undefined | Better HN

0 pointsqueuebert4y ago0 comments

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.

0 comments

wdroz4y ago

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

kwertzzz4y ago

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.

wdroz4y ago

You install cudatoolkit from conda then tensorflow with pip.

j / k navigate · click thread line to collapse

0 pointsqueuebert4y ago0 comments

0 comments

wdroz4y ago

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

kwertzzz4y ago

wdroz4y ago

You install cudatoolkit from conda then tensorflow with pip.

j / k navigate · click thread line to collapse