Unable install Pytorch from source for older GPU card

Hi there. I kinda combined the methods from this post and this one for my Windows 10, CUDA 10.0, GTX 950M environment.

I have done the following steps in Anaconda:

  1. conda activate fastai
  2. conda uninstall pytorch
  3. git clone --recursive GitHub - pytorch/pytorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration
  4. cd pytorch
  5. conda install -c pytorch pytorch

I checked the versions in my Jupyter Notebook by the following code:

import torch
print("PyTorch version: ", torch.__version__)
print("CUDA Version: ", torch.version.cuda)
print("cuDNN version is: ", torch.backends.cudnn.version())

Before the versions for fastai was:

PyTorch version: 0.3.1.post2
CUDA Version: 9.0
cuDNN version is: 7003

Now they are:

PyTorch version: 1.0.0
CUDA Version: 9.0
cuDNN version is: 7005
Device: cuda:0

Then tried the fastai course notebooks. Yep, it worked. The key is the PyTorch version. Thanks, guys.

P.S. My non-fastai environment is like:

PyTorch version: 1.0.0
CUDA Version: 10.0
cuDNN version is: 7401
Device: cuda:0