Skip to main content

Tensorflow and Pytorch for Cuda 10.0

And about the cuda, I believe you are setting up for Cuda 10.0 right? If that's the case I think the steps would be to install VS 2017, then CUDA and then download CUDNN (of compatible version) and copy it to the CUDA folder. also you might need to add cuda dir to system paths after copying cudnn.


Then:

conda create -n iq42mas python=3.6.5

conda install -c anaconda cudatoolkit=10.0

pip install tensorflow-gpu==1.15


conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch