PyTorch是GPU和CPU优化的深度学习张量库,学习python的都会接触到该库,很多人觉得PyTorch比Tensorflow更容易上手,今天就讲一下如何安装PyTorch?
工具/原料
1
PyTorch
2
电脑
方法/步骤
1
首先,安装该库就要到该库的官网上去找安装方法,打开官网后页面如下图所示
2
在官网页面向下划,滑到Get Started页面如下所示,run this command后面对应的命令即安装命令,右侧区域为电脑系统以及CUDA版本的选择,通过选择你电脑的配置来决定run this command后面对应的安装命令。
3
下图是我的电脑的配置,我选择的是Windows系统,CUDA9.1关于CUDA如何查看之前的一篇文章已经讲述了,可以参考我之前的文章,此处要根据你的CUDA版本选择,之后选择你的命令方式,我是使用pip命令的,Python的环境选择,根据的python环境配置来选择。
4
在新建的虚拟环境单击左键,弹出一个菜单,选择第一个open teminal,进入窗口命令程序。
5
打开命令窗口,输入pip3 install torchvision,然后回车,如下图所示开始安装
6
在如上界面等待20分钟之后,显示successfully,即安装成功,如下图所示。
注意事项
在第二步骤run this command后面显示的是两行命令,选择任意一行即可进行安装。
下一篇:Serv-U安装1-安装步骤