it seems that a fresh install doesnt necesarily install correct cuda version
I installed as http://dev.fast.ai/#Installing
but later I see that torch.cuda.is_available() returned false, but I have cuda… so I went to the page of pytorch and grab their command line
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done
## Package Plan ##
environment location: /home/tyoc213/miniconda3/envs/fastai2
added / updated specs:
- cudatoolkit=10.1
- pytorch
- torchvision
The following packages will be downloaded:
package | build
---------------------------|-----------------
cudatoolkit-10.1.243 | h6bb024c_0 347.4 MB
pytorch-1.5.0 |py3.7_cuda10.1.243_cudnn7.6.3_0 399.5 MB pytorch
torchvision-0.6.0 | py37_cu101 11.8 MB pytorch
------------------------------------------------------------
Total: 758.7 MB
The following packages will be DOWNGRADED:
cudatoolkit 10.2.89-hfd86e86_1 --> 10.1.243-h6bb024c_0
pytorch 1.5.0-py3.7_cuda10.2.89_cudnn7.6.5_0 --> 1.5.0-py3.7_cuda10.1.243_cudnn7.6.3_0
torchvision 0.6.0-py37_cu102 --> 0.6.0-py37_cu101
Proceed ([y]/n)? y
Downloading and Extracting Packages
pytorch-1.5.0 | 399.5 MB | ########################################################### | 100%
cudatoolkit-10.1.243 | 347.4 MB | ########################################################### | 100%
torchvision-0.6.0 | 11.8 MB | ########################################################### | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
Like you see it installed incorrect cuda version, here it is 10.1 not 102, now torch.cuda.is_available() is true.
I dont know if this was just a failure or why it installed those other libs. Or is there a way that it dectects the correct ones on install? if not, ppl will think some things are “slow” because using CPU.