更新:更新安装指南增加新版官网安装说明
This commit is contained in:
parent
fb61056926
commit
fba6ab225b
@ -10,6 +10,7 @@ anaconda安装完成后可以开始创建环境,这里以win10 系统为例。
|
||||
conda create -n pytorch python=3.6
|
||||
# 切换到pytorch环境
|
||||
activate pytorch
|
||||
# ***以下为1.0版本安装***
|
||||
#安装GPU版本,根据cuda版本选择cuda80,cuda92,如果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` 这样才能使用镜像源。
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user