如何安装Pytorch

1 什么是Pytorch

pytorch是一个基于Python的科学计算包,其类似于Numpy但是能利用GPU加速,也是一个非常灵活和快速用于深度学习的研究平台。

2 安装Pytorch

检查自己的电脑是否有NVIDIA独立显卡

从硬件层面上说,NVIDIA从Geforce9XXX系列显卡,也就是G92/G94/G96核心开始支持CUDA技术。但NVIDIA通过在驱动层面上的优化,让从Geforce8XXX系列显卡开始往后的产品,全部支持CUDA通用计算技术。也就是说从G80/G84/G86核心开始,往后的产品都支持CUDA技术。

在运行中启动dx诊断工具查看自己的显卡型号。

如何安装Pytorch_第1张图片

如何安装Pytorch_第2张图片

确认自己的cuda版本

如何安装Pytorch_第3张图片

在Pytorch的官网查询适合自己的安装指令

链接:https://pytorch.org/get-started/locally/

选择需要的版本,运

你可能感兴趣的:(PyTorch图像分类教程,pytorch,python,深度学习)