Pytorch框架的安装

2019-02-27  本文已影响0人  鲜芋牛奶西米爱solo

网址:https://pytorch.org/get-started/locally/

我的安装选项:pytorch build:stable1.0 + OS:Windows + Package:pip + Python3.7 + CUDA:9.0

Command:

                pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp37-cp37m-win_amd64.whl

                pip3 install torchvision

执行命令1 执行命令2

#torchvision主要集成了一些数据集,dl模型等

安装成功

上一篇下一篇

猜你喜欢

热点阅读