visual studio code配置pytorch环境

首先，你需要安装 Anaconda 或者 Miniconda。这两个软件包含了我们需要的 Python 环境和一些常用的科学计算库。

安装好后，打开 Anaconda Prompt 或者终端，输入以下命令来创建一个新的 Python 环境：

conda create --name pytorch_env python=3.7

这里我们创建了一个名为 pytorch_env 的环境，并指定 Python 版本为 3.7。等待环境创建完成后，可以通过以下命令来激活这个环境：

conda activate pytorch_env

接下来，我们需要安装 PyTorch 和 torchvision。在 Anaconda Prompt 或者终端中输入以下命令：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

其中，cudatoolkit 参数用于指定 CUDA 工具包的版本。如果你的电脑没有 Nvidia 显卡，可以将它去掉。

安装完成后，你就可以在 Visual Studio Code 中使用 PyTorch 了。在 VS Code 中打开一个 Python 文件，按下 Ctrl + Shift + P，输入“Python: Select Interpreter”，选择刚才创建的 pytorch_env 环境即可。