1 PyTorch的安装方法选择
安装最经典的方法是在官网选择电脑的系统,GPU型号,PyTorch版本后,将生成的安装代码直接复制并使用命令行安装。
但由于网络的原因,相信大家基本在安装的时候都遇到过各种而样的问题导致安装失败。所以一般会使用镜像安装和离线安装代替
其中离线安装是最不会产生网络波动导致安装失败的方法,在此推荐使用。
2 PyTorch安装包的选择
PyTorch安装包的下载链接
打开后会看到所有的发布过的安装包,大多教程都没有对如何选择安装包进行讲解,导致在安装的时候出现了安装包不匹配等类似的报错,此处对包的选择做一些说明。
离线安装使要下载torch和torch version两个文件,以torch为例,以cpu开头的是CPU版本,以CU开头的是GPU版本,CU后面接的序号就是GPU系统的型号。比如我的是11.7版本的CUDA,就要选择CU117开头的安装包。
显卡版本后面是对PyTorch版本的选择,这个一般使用最新版本即可,版本
后面的%2代表beta版本。
版本后面的cp××代表Python版本
cp后面的liunux/win代表系统电脑版本
因此,我的电脑是11.7版本的CUDA,3.9版本的Python,Windows64位系统,希望下载1.13.1版本的PyTorch。所以我下载了下面这两个文件:
cu117/torch-1.13.1%2Bcu117-cp39-cp39-win_amd64.whl
之后找到对应的torch version进行下载。
cu117/torchvision-0.14.1%2Bcu117-cp38-cp38-win_amd64.whl
之后打开cmd/Anconda prompt/Pycharm的Terminal进行安装。
3 PyTorch安装
首先将操作目录调整到对应的系统盘
比如此时的操作目录在c盘桌面,而安装包在d盘,输入
d:
即可将运行路径切换到d盘,随后使用cd将运行路径切换到对应的文件夹,以我个人的文件夹路径为例
cd D:\Python\Anaconda3\lib\site-packages
最后执行安装命令
pip install torch-1.13.1+cu117-cp39-cp39-win_amd64.whl
先安装torch,再安装torchversion,安装包的名称使用下载下来之后的安装包文件名称,直接复制过来就好。
两个包安装完之后就大功告成啦,入过之后运行需要安装torch的其他包,同样的方法找到对应的安装包版本进行安装即可。