问题展示:
我先前在pytorch官网用conda安装pytorch但是太慢,我就用pip安装
我在我建的conda环境:pytorch中运行这段命令,很快success
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
然后我在pytorch环境中输入conda list检查,发现找不到我安装的包,接着我输入pip list是可以找到我刚刚安装的包的。
我输入python,接着import torch,报错ModuleNotFoundError: No module named 'torch'
解决方案:
1. 先在pytorch环境下把一开始安装的pytorch卸载了:
pip uninstall torch torchvision torchaudio
2. 把原来的pip3 install改成python -m pip install,即:
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
检查pytorch: