安装pytorch时,文件名称的识别

您所在的位置:网站首页 文件夹指什么意思和新含义一样 安装pytorch时,文件名称的识别

安装pytorch时,文件名称的识别

2024-07-15 20:42| 来源: 网络整理| 查看: 265

这里记录一下在安装pytorch时,下载whl文件时,选择版本的问题。 首先提一下pytorch安装。 有显卡的话,是需要先安装cuda的,cuda的安装是需要根据你自己电脑的显卡型号来安装,具体可百度。然后根据cuda型号和自己电脑的系统和python版本来下载安装pytorch。 没有显卡的话,就只能安装cpu版本的pytorch。这就只需要根据自己电脑的系统和python的版本来下载安装pytorch。 首先到官网https://pytorch.org/,往下面拉,可以看到让你选择配置。 你可以在选择配置后,根据Run this Command里自动生成的代码,在命令行里进行下来。但是一般情况下,都非常慢或者出现很多问题,即使是换了国内的源。 在这里插入图片描述 所以建议直接下载whl文件进行安装。 网址:https://download.pytorch.org/whl/torch_stable.html 在这个网站下载所需要版本的torch和torchvision,然后再命令行里pip install 进行安装就行了。 下面说明这些文件名称的含义。 在这里插入图片描述 最前面的cpu就是cpu版本的意思,如果要gpu版本的,前面就是cu开头,cu后面的数字就是cuda的版本号,你自己用的什么版本的cuda就下载哪个版本的。 查询cuda版本,在命令行输入nvcc -V 在这里插入图片描述 这里的cuda版本是10.0,那么我就应该选择cu100开头的文件。在这里插入图片描述 再后面的torch或者torchvision就表示是哪个文件了,后面跟的就是这个文件的版本,建议下载最新版本。 再后面cp+数字就是对映python的版本,如果你的python是3.7的,就下载cp37的文件。 再后面,就是操作系统,win或者mac或者linux。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3