pytorch cudatoolkit

這一篇 說， pytorch 有自己的 cuda lib，所以conda 安裝時，指定要用的 cuda 版本 (可以用的)。

 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

裝完，測試看：

>>> import torch
prin>>> 
>>> print(torch.cuda.is_available())
True

初步看起來 OK


另外，這一篇 的說明更完整(多):

In [1]: import torch

In [2]: torch.cuda.current_device()
Out[2]: 0

In [3]: torch.cuda.device(0)
Out[3]: 

In [4]: torch.cuda.device_count()
Out[4]: 1

In [5]: torch.cuda.get_device_name(0)
Out[5]: 'GeForce GTX 950M'

In [6]: torch.cuda.is_available()
Out[6]: True