Pytorch安装
决定入手pytroch。。
https://pytorch.org/resources,打开,
注意:之前就装了CUDA9.0,因此整个安装过程只有一行代码:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安装完成,测试一下
>>> import torch >>> print(torch.__version__) 1.1.0 >>> print('gpu', torch.cuda.is_available()) gpu True
没毛病,正式开启DL之旅
人生苦短,何不用python