如何安装Pytorch

Posted 算法与编程之美

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何安装Pytorch相关的知识,希望对你有一定的参考价值。

1 什么是Pytorch

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

2 安装Pytorch

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

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

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

确认自己的cuda版本

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

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

选择需要的版本,运

以上是关于如何安装Pytorch的主要内容,如果未能解决你的问题,请参考以下文章

如何解决conda安装pytorch,总是下载失败的问题。

如何利用C++来调用PyTorch训练好的模型

如何利用C++来调用PyTorch训练好的模型

深度学习环境搭建anaconda+pycharm+pytorch

如何配置及安装pytorch虚拟环境

如何安装pytorch