pytorch安装的超级详细教程(没有之一)

一、发展历程 （简单介绍）

(15年)caffe --> (16年)tensorflow1.x --> (17年)keras --> (18年)Tensorflow2.x --> (19年)pytorch。

面向gihub开源项目编程。

向下支持比较好，各个版本之间支持比较好，兼容性强。

版本分为GPU和CPU两种

二、配置环境（cpu）（比较容易）

pip install -U torch -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

pip install -U torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

三、配置环境（GPU）（比较复杂）

1.查看自己的卡适合哪个版本的CUDA

cmd命令如下

nvdia-smi

 建议：优先装11.x，因为该公司在更新的时候，可能对10.x版本就不再进行支持了。

2.安装CUDA

直接在百度搜CUDA（下面这篇博客详细的介绍了如何安装CUDA,亲测有效）

手把手教你安装CUDA（一看就会）_cuda安装_时代&信念的博客-CSDN博客

安装之后，在cmd中输入

nvcc-V

出现下列界面，证明安装成功

2.安装pytorch

1）直接在百度搜pytorch

PyTorch

然后我们在下面的这个网站里面来配置pytorch

https://download.pytorch.org/whl/torch_stable.html

按照自己的cuda 和python的版本去下载torch、torchvistion

1、2、3、4分别代表的是CUDA的版本、python版本、操作系统的类型、操作系统的位数。

将上面的两个文件安装到anacoda的sciprits下面

在这个'D:\anacoda\Scripts>'路径下pip 上面的包

torchvision-0.11.0+cu113-cp37-cp37m-win_amd64.whl

pip install torch-1.10.2+cu113-cp37-cp37m-win_amd64.whl

 上面的命令要根据自己下载的torch和torchvision进行改变

2）检查是否安装成功：

输入下列命令：

python

print(torch.cuda.is_available())

若出现下面的’TRUE‘，则证明安装成功

至此pytorch配置完成。