pytorch的CPU版本&GPU版本的环境安装(附带cuda和cudnn)

一、Windows(64bit)下安装pytorch-cpu

首先说个题外话,为了日后使用的方便最好是将将conda和pip的软件源修改成清华的源,这样的话,使用conda或者pip安装软件速度会快很多。可使用如下指令进行设置:(当然只是建议,与本次pytorch的安装无关,pytorch都是直接从官网上下载,速度还挺快)

conda  config  --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda  config  --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda  config  --set show_channel_urls yes

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

1.1 使用pip安装

以Python 3.6为例

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl

pip3 install torchvision

1.2 使用anaconda安装

评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值