查看torch版本_No module named 'torch'

运行代码显示：ModuleNotFoundError: No module named 'torch'

嗯？环境中没有torch。。。

---

• 本地安装

查看匹配的环境和安装代码：

PyTorch​pytorch.org

打开Anaconda prompt

conda env list # 列出环境
conda activate <环境> # 切换到环境
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
 

---

• 服务器上

查看cuda版本：

ls -l /usr/local | grep cuda

显示：

lrwxrwxrwx 1 root root    9 Sep 30  2019 cuda -> cuda-10.1
drwxr-xr-x 1 root root 4096 Sep 30  2019 cuda-10.1

安装pytorch：cudatoolkit=10.1

conda install pytorch torchvision cudatoolkit=10.1 -c pytorc

y继续

安装完毕。