pytorch的炼丹之路-环境配置

本文详细介绍了在Windows10系统中安装PyTorch GPU版本的步骤,包括检查GPU支持、从官网选择适合的版本、激活虚拟环境、使用pip安装指定版本的PyTorch、验证GPU运行以及在Jupyter中设置和使用PyTorch环境。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  1. 我的电脑是 windows10系统,在任务管理器中 查看是否有GPU
    在这里插入图片描述
    2.去官网 按照对应的命令下载即可,
    https://pytorch.org/get-started/locally/
    CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台
    我是这么选择的,

在这里插入图片描述
这里有一个关于Windows(64bit)下安装pytorch-gpu
https://www.huaweicloud.com/articles/7c2f84bc1bc86dec1c7f30561469ec64.html 的链接,我也不知道问什么要安装那么多东东。我没装,因为我输入 nvcc -V 啥也不是
在这里插入图片描述

3.激活虚拟环境,开启安装,
pip3 install torch1.9.0+cu102 torchvision0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述

pip install -i https://pypi.douban.com/simple ipykernel 为了将 该pytorch 的虚拟环境添加到jupyter中,

(pytorch_nlp) D:\all_env\pytorch_nlp\Scripts>python -m ipykernel install --user --name=pytorch_nlp
在这里插入图片描述

然后可以查看一下 安装了哪些虚拟环境kernel ,
在这里插入图片描述

  1. 启动jupyter ,切换环境在这里插入图片描述
  2. 将官网中的代码复制,运行一下,是运行在GPU上的。。
    在这里插入图片描述
    6.以上就是pytorch GPU的安装,
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

weixin_43351935

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

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

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

打赏作者

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

抵扣说明:

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

余额充值