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之旅

 

posted @ 2019-08-01 20:21  嶙羽  阅读(331)  评论(0编辑  收藏  举报