【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】

pytorch

学习需要安装pytorch,记录一下自己的安装过程!

查看Python版本

在这里插入图片描述

先查看CUDA版本

nvidia-smi

在这里插入图片描述
如图我的是11.6版本的

在线安装进入官网下载

在这里插入图片描述
选择对应的版本和安装方式!
但是下载速度慢,所以可以离线安装

离线下载安装

pytorch与torchvision下载链接:
https://download.pytorch.org/whl/torch_stable.html

选择对应版本下载
例如:
在这里插入图片描述

CUDA没有11.6的我选择11.5,Python版本3.8,Windows64位

下载完pytorch和torchvision的.whl文件之后,在下载的目录内使用下列命令即可:

pip install torch-1.11.0+cu115-cp38-cp38-win_amd64.whl
pip install torchaudio-0.11.0+cu115-cp38-cp38-win_amd64.whl

在这里插入图片描述
安装成功!

是否安装成功

运行代码看能否使用

import torch

在这里插入图片描述
运行成功🎉🎉🎉🎉!

  • 2
    点赞
  • 10
    收藏
  • 打赏
    打赏
  • 1
    评论

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:博客之星2021 设计师:Hiro_C 返回首页
评论 1

打赏作者

南蓬幽

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

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

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

打赏作者

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

抵扣说明:

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

余额充值