PyTorch环境的配置

PyTorch的配置

这里附上官网的链接https://pytorch.org/get-started/locally/

CPU版本

选择PyTorch版本(PyTorch Build)，Stable(稳定版)，Nightly(测试版)
系统版本，这里只介绍Windows系统的PyTorch环境配置
Package(包管理器)用Anaconda的就选conda，普通Python环境的就pip
Language当然选Python
Conpute Platform（计算平台）安装CPU版本就选择CPU
下面Run This Command：就是这里会给出安装命令
如果有设置Path路径就可以在命令台运行安装命令，然后等待安装结束就行

GPU版本

现在PyTorch的GPU版本只支持英伟达(nvidia)的GPU
可以在命令台输出nvdia-smi来查看显卡最高支持的cuda版本

现在PyTorch 1.9最高支持CUDA11.1
可以在Nvidia官网下载CUDA Toolkit
以CUDA11.1为例

这里附上链接https://developer.nvidia.com/cuda-toolkit-archive下载CUDA11.1进行安装
安装过程中一定要选择自定义安装，最好就只安装CUDA组件，因为CUDA Toolkit中包含了驱动程序等其他的内容(而且一般版本比较老)可能会覆盖你当前的版本造成异常
然后下一步下一步就行了。
最后就是在Compute Platform选上CUDA11.1然后根据给出的公式在命令台输入安装即可