pytorch 和python对应关系 pytorch1.7对应python

转载

ctaxnews 2023-11-21 21:23:59

文章标签 pytorch 和python对应关系 pytorch pillow numpy conda 文章分类 PyTorch 人工智能

我在Anaconda下创建的新环境为

python 3.7.0

pytorch 1.8.0

pillow 9.5.0

numpy 1.21.5

能够正常运行

如果我这个版本够用的话可以按照这个版本进行安装

具体步骤如下：

1.在Anaconda Prompt创建python3.7.0版本的新环境

conda create -n pytorch37 python==3.7

上述代码是创建一个名为pytorch37的新环境，新环境中的python版本等于3.7.0(Anaconda中的3.7默认为3.7.0)

2.去如下pytorch官网上找自己的Python所对应的版本

pytorch 和python对应关系 pytorch1.7对应python_conda

从百度或者其他搜索引擎进入 PyTorch 的官网，往下拉一点可以看到这样的一部分，下面有个

Previous versions of PyTorch，单击进入

可以根据你在其他地方获取的python对应的pytorch版本进行选择；这里我提前在搜索引擎上查找了我的 python3.7.0可以对应 pytorch 的1.8.0版本，因此下拉找到对应的 Anaconda 指令

pytorch 和python对应关系 pytorch1.7对应python_numpy_02

为方便我把 Linux and Windows 放在下面供你们参考，省去你们一些步骤

# CUDA 10.2
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

注意这里根据你是要使用 GPU 版本的 PyTorch还是 CPU 版本的 PyTorch需要选择合适的命令进行运行。前两个为 CUDA 版本的 PyTorch，即 GPU 版本，因为使用 GPU 需要通过 CUDA 来调用GPU，而 CPU 是可以被程序直接唤起的。如果你的电脑是集成显卡，那很高兴你跟我第一台电脑是一样的，不需要去管显卡的事情了。

这里给新手朋友说下如何查看是集显还是独显

首先打开你的控制面板，直接在搜索框搜索"设备管理器"

pytorch 和python对应关系 pytorch1.7对应python_conda_03

打开设备管理器之后点开显示适配器，可以看到有 NVIDIA 的垃圾独显 MX150 (老电脑了)，虽然性能差，但好歹能有与之匹配的 CUDA；当然如果没有 NVIDIA 的显卡，那说明你的电脑不支持CUDA，或者是集成显卡

这里给各位想用 GPU 版本的伙伴说一下如何查看自己的 CUDA 兼容版本。由于只有NVIDIA英伟达显卡有 CUDA 适配，因此非英伟达显卡的伙伴们直接调用上面 CPU Only 后面的命令即可，CUDA 的内容暂时与你无缘，老老实实用 CPU 版本做点简单的东西吧，CPU 版本的算力能跑的动Resnet 残差网络就已经差不多了

首先在你电脑中搜索 NVIDIA 会出现如下应用

pytorch 和python对应关系 pytorch1.7对应python_numpy_04