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

简介: 首先查看自己的Python版本命令行cmd输入Python即可查看然后先查看CUDA版本进入cmd输入以下 nvidia-smi

pytorch

人工智能学习需要安装pytorch,今天记录一下自己的安装过程!

首先查看自己的Python版本

命令行cmd输入Python即可查看
结果如下:
在这里插入图片描述
由此可以看出Python为3.8版本

然后先查看CUDA版本

进入cmd输入以下

nvidia-smi

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

在线安装进入官网下载

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

离线下载安装

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   # 能否调用pytorch库

print(torch.cuda.current_device())   # 输出当前设备(我只有一个GPU为0)
print(torch.cuda.device(0))   # <torch.cuda.device object at>
print(torch.cuda.device_count())  # 输出含有的GPU数目
print(torch.cuda.get_device_name(0))  # 输出GPU名称 --比如1080Ti
x = torch.rand(5, 3)
print(x)  # 输出一个5 x 3

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

目录
相关文章
|
24天前
|
PyTorch Linux 算法框架/工具
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
这篇文章是关于如何使用Anaconda进行Python环境管理,包括下载、安装、配置环境变量、创建多版本Python环境、安装PyTorch以及使用Jupyter Notebook的详细指南。
198 1
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
|
2月前
|
iOS开发 MacOS Windows
Mac air使用Boot Camp安装win10 ,拷贝 Windows 文件时出错
Mac air使用Boot Camp安装win10 ,拷贝 Windows 文件时出错
|
22天前
|
数据安全/隐私保护 Windows
安装 Windows Server 2019
安装 Windows Server 2019
|
22天前
|
Windows
安装 Windows Server 2003
安装 Windows Server 2003
|
23天前
|
NoSQL Shell MongoDB
Windows 平台安装 MongoDB
10月更文挑战第10天
33 0
Windows 平台安装 MongoDB
|
27天前
|
Windows Python
Windows安装dlib,遇到问题汇总解决
Windows安装dlib,遇到问题汇总解决
29 4
|
24天前
|
机器学习/深度学习 缓存 PyTorch
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
这篇文章是关于如何下载、安装和配置Miniconda,以及如何使用Miniconda创建和管理Python环境的详细指南。
286 0
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
|
30天前
|
Oracle 关系型数据库 MySQL
Mysql(1)—简介及Windows环境下载安装
MySQL 是一个流行的关系型数据库管理系统(RDBMS),基于 SQL 进行操作。它由瑞典 MySQL AB 公司开发,后被 Sun Microsystems 收购,现为 Oracle 产品。MySQL 是最广泛使用的开源数据库之一,适用于 Web 应用程序、数据仓库和企业应用。
52 2
|
1月前
|
JavaScript Windows
windows安装vue
windows安装vue
|
1月前
|
应用服务中间件 Apache Windows
免安装版的Tomcat注册为windows服务
免安装版的Tomcat注册为windows服务
80 3