PyCharm中conda环境安装PyTorch 1.6.0 CPU版失败求助

问题根源 #

PyTorch 1.6.0 和 torchvision 0.7.0 官方仅支持 Python 3.6-3.8，你的Python 3.9不在兼容范围内，这是导致conda包冲突和pip找不到版本的核心原因。

解决步骤 #

1. 创建兼容的Python版本conda环境

先新建一个Python 3.8（或3.6/3.7）的conda环境，这是最稳妥的方式：

conda create -n torch160_env python=3.8
conda activate torch160_env

2. 用conda安装（优先推荐）

在激活的环境中执行官方conda命令，此时版本匹配，不会出现zlib冲突：

conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch

3. 备选：用pip安装

如果conda源出现问题，在激活的环境中执行以下pip命令，确保环境Python版本为3.6-3.8，就能找到对应whl包：

pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

4. 验证安装

运行以下代码确认安装成功：

import torch
import torchvision
print(torch.__version__)  # 输出1.6.0
print(torchvision.__version__)  # 输出0.7.0
print(torch.cuda.is_available())  # CPU版输出False

5. 在PyCharm中使用该环境

打开PyCharm的Settings > Project: [你的项目名] > Python Interpreter，点击齿轮图标选择Add，然后选Conda Environment，找到刚创建的torch160_env环境即可。

内容的提问来源于stack exchange，提问作者Taarun Srinivas