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:
- conda activate fastai
- conda uninstall pytorch
- git clone --recursive GitHub - pytorch/pytorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration
- cd pytorch
- 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