【win11安装Anaconda和GPU PyTorch】win11安装GPU PyTorch

  1. 安装NVIDIA驱动
    下载地址
    点击开始搜索,自动搜索自己电脑显卡的型号,产品型号记得选择带notebooks的;
    点击立即下载,安装包下载完,按默认配置安装即可。
    在这里插入图片描述
  2. 安装Anaconda
    下载地址
    可能需要先注册一个账号,安装包下载完,按默认配置安装即可。
    在这里插入图片描述
  3. 进入 Anaconda prompt 命令窗口
    先使用清华源,必须使用http
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
    conda config --set show_channel_urls yes
    
    打开C:\Users\你的用户名 文件夹下修改.condarc文档,将其中的内容替换为如下代码:
    channels:
      - defaults
    show_channel_urls: true
    default_channels:
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
      - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
    custom_channels:
      conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    
    回到 Anaconda prompt 命令窗口
    // PyTorch是名字,可以改,必须指定python版本号
    conda create -n PyTorch python=3.9
    // 查看 Anaconda 的所有环境
    conda info --envs
    
    在这里插入图片描述
  4. 进入pytorch官网
    官网地址
    在这里插入图片描述
    选择相应配置,复制配置下面的命令
  5. 回到 Anaconda prompt 命令窗口
    输入下面的命令进入PyTorch环境
    conda activate PyTorch
    
    在这里插入图片描述
    输入刚刚复制的命令,等待下载完成
    conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
    
  6. 测试安装成功
    重新进入 Anaconda prompt 命令窗口
    // 进入python环境
    python
    // 测试
    import torch
    torch.cuda.is_available()
    
    输出为
    True
    
    至此安装成功
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Katherine121

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值