如何安装pytorch?

作者:清风拂面 | 创建时间: 2023-07-02
PyTorch是GPU和CPU优化的深度学习张量库,学习python的都会接触到该库,很多人觉得PyTorch比Tensorflow更容易上手,今天就讲一下如何安装PyTorch?...
如何安装pytorch?

操作方法

首先,安装该库就要到该库的官网上去找安装方法,打开官网后页面如下图所示

在官网页面向下划,滑到Get Started页面如下所示,run this command后面对应的命令即安装命令,右侧区域为电脑系统以及CUDA版本的选择,通过选择你电脑的配置来决定run this command后面对应的安装命令。

下图是我的电脑的配置,我选择的是Windows系统,CUDA9.1关于CUDA如何查看之前的一篇文章已经讲述了,可以参考我之前的文章,此处要根据你的CUDA版本选择,之后选择你的命令方式,我是使用pip命令的,Python的环境选择,根据的python环境配置来选择。

在新建的虚拟环境单击左键,弹出一个菜单,选择第一个open teminal,进入窗口命令程序。

打开命令窗口,输入pip3 install torchvision,然后回车,如下图所示开始安装

在如上界面等待20分钟之后,显示successfully,即安装成功,如下图所示。

温馨提示

在第二步骤run this command后面显示的是两行命令,选择任意一行即可进行安装。
点击展开全文

更多推荐