更新:更新安装指南增加新版官网安装说明

This commit is contained in:
zergtant 2019-02-28 20:51:17 +08:00
parent fb61056926
commit fba6ab225b

View File

@ -10,6 +10,7 @@ anaconda安装完成后可以开始创建环境这里以win10 系统为例。
conda create -n pytorch python=3.6
# 切换到pytorch环境
activate pytorch
# ***以下为1.0版本安装***
#安装GPU版本根据cuda版本选择cuda80cuda92如果cuda是9.0版,则不需要
#直接conda install pytorch -c pytorch 即可
# win下查看cuda版本命令nvcc -V
@ -19,6 +20,20 @@ conda install pytorch cuda92 -c pytorch
# torchvision 是torch提供的计算机视觉工具包后面介绍
pip install torchvision
# *** 官方更新了1.01 所以安装方式也有小的变更
# torchversion提供了conda的安装包可以用conda直接安装了
# cuda支持也升级到了10.0
# 安装方式如下:
# cpu版本
conda install pytorch-cpu torchvision-cpu -c pytorch
# GPU版
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
# cudatoolkit后跟着相应的cuda版本
# 目前测试 8.0、9.0、9.1、9.2、10.0都可安装成功
```
需要说明的一点是如果使用清华源可以直接添加pytorch源镜像去掉并且去掉`-c pytorch` 这样才能使用镜像源。