本文最后更新于 2025-01-24,文章内容可能已经过时。

有很长一段时间没有在Windows上配置过环境了,之前没有记录,导致这次配置时出了点小问题。

安装Pytorch

注意安装Pytorch如果是gpu版本不要使用pip清华源,清华源只有cpu版本

PyTorch

笔者使用官网的conda安装方式:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

import torch failed

conda list中我找到了pytorch,但是使用jupyter时却无法import

随后我在命令行尝试了python3python 导入torch,便发现了端倪

使用python命令可以导入,而使用python3命令不能导入,jupyter默认使用python3运行,那么问题在这里了

我们可以看到python3只有系统默认的解释器,没有conda环境的,于是我们找到conda环境python解释器的位置D:\softwares\conda\envs\torch\python.exe ,将其复制一个副本并重命名为python3.exe ,此时我们在conda中执行python3 ,用的便是这个安装了torch的解释器

在jupyter中使用conda环境

Jupyter notebook 指定 Python 解释器

jupyter中我们还需要导入ipykernel

conda activate torch
pip install ipykernel -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
python -m ipykernel install --name torch
jupyter notebook

那让我们在jupyter中测试一下: