下载安装pytorch

安装pytorch过程
  • 首先安装anaconda

  • 创建环境

  • pytorch安装

  • 检验安装

 1.安装anaconda(我的电脑已经安装过anaconda,所以这里跳过)

     如果没有安装,可以参考     Anaconda超详细安装教程(Windows环境下)_conda安装-CSDN博客​​​​​​

2.创建环境(pytorch)

    首先打开Anaconda Prompt

    

   
然后创建环境(环境名为pytorch,当前的环境python=3.9.7)

   

    是否继续(y 继续)

   
验证环境是否创建成功,输入activate pytorch(pytorch是你定义的这个环境的名字),左边      的环境就从base(基本环境),变成了pytorch环境

    

3.pytorch安装

   
 查看CUDA版本数,进入命令符号窗口,输入nvidia-smi,查看当前驱动的版本号,观察                   Driver Version的值是否大于400,如果小于请更新显卡驱动

     

   
进入pytorch官网 PyTorch

    

 
  选择适应的版本(查看自己的CUDA版本数,我的cuda是10.2)    

    

     

   

   
 复制这段指令(由于从官网下载过慢,容易导致失败,我换成了清华源)

     pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f                       https://download.pytorch.org/whl/cu102/torch_stable.html -i                                                             https://pypi.tuna.tsinghua.edu.cn/simple

     

4.检验安装

   先输入python,然后输入import torch,如果输入后没有任何报错,然后再输入                               torch.cuda.is_available(),返回的是True,就成功了

   

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值