这里记录一下在安装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。
安装pytorch时,文件名称的识别
最新推荐文章于 2024-01-05 10:08:42 发布