安装pytorch-gpu版本(分享我的成功案例)

用官网的命令太慢,换国内的源也不行,因为国内的源会默认给你安装cpu版本,所以,,,还是往下看吧

一、如果还没安装Anaconda和Cuda的先去这个链接(Windows 10安装Tensorflow-gpu及cuda+cuddn详细图解(刚装完赶紧做笔记:作经验分享)_我系渣渣的博客-CSDN博客)安装前三步后再来安装Pytorch。

二:创建一个独有的虚拟环境

     1、创建 conda 环境

           启动 Anaconda prompt 

conda create -n xxxx pip python=3.6    #xxxx是你虚拟环境的名,想要啥就啥,python版本号可换

     2、激活创建的 conda 环境,进入创建的虚拟环境

            conda activate xxxx

二、开始安装pytorch-gpu

  • 这里建议先离线下载好再安装,因为用官网的命令安装总是容易安装成cpu版本的,我们到这个网站 ​​​​​​https://download.pytorch.org/whl/torch_stable.html 下载好所需要的包,然后pip安装
  • 也可以到打开清华源下的pytorch库
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  • 找到自己合适的版本,比如我需要GPU版本的并且支持cuda10,网页ctrl+f快捷搜索,找到文件中
 - pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2

       版本号:1.2.0    ------根据自己需求更改

  • nightly版本可以看作是开发版,一般不选用,
    命名格式为
     pytorch-版本-python版本-cuda表示gpu版-cudnn版本.tar.bz2大小一般为5、600M

  • 下载后,放到合适的盘,接着在刚才创建的那个虚拟环境下,cd到文件目录
    输入命令

  • conda install --offline pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2 #文件全称
    

    安装完pytorch后,直接输入命令安装完依赖的包:

  • conda install torchvision
    

    打开conda中的python进行测试

  • import torch
    print(torch.cuda.is_available())
    print(torch.version.cuda)
    

    输入结果为True和10.0就对了

  • 4
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值