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
原文地址:https://www.cnblogs.com/yqpy/p/11285064.html