安装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

      3、查看虚拟环境列表

        我们通过cmd或者Anaconda Prompt可以打开命令列表,两者都行

                conda env list
                conda info --envs

        4、删除环境

                conda remove -n xxxxx(名字) --all

二、开始安装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 #文件全称
    

    第二个在线安装方法:

  • 镜像地址:
    清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
    
    # 输入安装命令
    conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
    
    conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
    
    conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
    
    conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
    
    # 经过测验,以上Pytorch的版本输入安装命令会自动替换为cpu版本的下载,1.10.1版本下载的是gpu的
    conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
    

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

  • conda install torchvision
    

    打开conda中的python进行测试

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

    输入结果为True和10.0就对了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值