如何安装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的主要内容,如果未能解决你的问题,请参考以下文章