PyTorch v_9_9, Windows v v-xxxxx

阅读量：

PyTorch 1.5环境配置指南

点击访问官方网站 https://pytorch.org/

通过pip进行安装

使用Python3.7及以上版本安装CPU版本的pip

复制代码

    pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

Python3.7及以上版本通过pip安装GPU版本
当前GPU版本兼容的CUDA版本包括8.0、9.0以及10.0，用户需根据自身需求选择合适的版本进行下载与安装，切勿同时运行以下三个命令！偏好使用conda环境的用户可自行选择

复制代码

    pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

使用conda进行软件安装

复制代码

    conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

不得不说，相较于tensorflow的安装过程，此方法的便捷性提升了千倍之多！

全部评论 (0)

还没有任何评论哟~