在Anaconda里创建的虚拟环境中成功安装了pytorch,键入python以及import torch,显示“No module named 'torch' ”。
但在base环境中,import torch却有用。这是为什么呢?

在Anaconda里创建的虚拟环境中成功安装了pytorch,键入python以及import torch,显示“No module named 'torch' ”。
但在base环境中,import torch却有用。这是为什么呢?

上午好☀️☀️☀️️
本答案参考通义千问
你遇到的问题是:在Anaconda创建的虚拟环境中安装了PyTorch,但无法导入torch模块,而在base环境中却可以正常导入。
这说明你的虚拟环境并没有正确安装PyTorch,或者安装的PyTorch没有被正确识别。
未激活虚拟环境
安装命令错误
pip install torch而不是conda install pytorch),导致安装位置与当前环境不一致。路径问题
环境隔离问题
在终端中运行以下命令:
which python
或在Python中运行:
import sys
print(sys.executable)
base 环境的路径(如 /opt/anaconda3/bin/python),说明你没有激活虚拟环境。/opt/anaconda3/envs/your_env_name/bin/python),则说明你已经激活了正确的环境。在终端中激活你的虚拟环境:
conda activate your_env_name
注意:将
your_env_name替换为你实际的虚拟环境名称。
在激活后的虚拟环境中运行:
pip show torch
或者:
conda list | grep torch
conda 安装(推荐)conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
注意:
cudatoolkit=11.8是示例版本,根据你的CUDA版本选择对应的版本号。
pip 安装(需确保在虚拟环境中)pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
注意:根据你的CUDA版本调整版本号。
在虚拟环境中运行以下代码:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果输出如下内容,说明安装成功:
1.13.1
True
| 步骤 | 操作 |
|------|------|
| 1 | 确认当前使用的Python环境 |
| 2 | 激活正确的虚拟环境 |
| 3 | 检查是否安装了PyTorch |
| 4 | 使用 conda 或 pip 重新安装PyTorch |
| 5 | 验证是否安装成功 |
conda 安装PyTorch通常更稳定,特别是对于CUDA支持。如果你愿意提供具体的虚拟环境名称、Python版本以及安装命令,我可以进一步帮你排查问题!