win10 pytorch0.4.0-GPU安装

在已安装cuda、cudnn基础上,简要记录一下安装流程:

(1)以管理员方式运行Anaconda Prompt,创建一个pytorch环境:

        conda create -n pytorch python=3.5 numpy pyyaml mkl

(2)激活pytorch环境:activate pytorch

(3)增加一个镜像源:

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

     conda config --set show_channel_urls yes           //设置是否在安装的时候显示该链接

(4)之前用的pip版本是9.0.1,更新到10.0.1,输入命令:

     python -m pip install --upgrade pip

(5)安装pytorch 0.4.0 gpu,python版本是3.5.4,输入命令:

     pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl

有可能存在网络问题,尝试多次后终于成功。

(6)安装torchvision:

    pip install torchvision

(7)测试:

>>> import torch
>>> a = torch.Tensor([1.])
>>> a.cuda()
tensor([ 1.], device='cuda:0')
>>> from torch.backends import cudnn
>>> cudnn.is_acceptable(a.cuda())
True


评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值