vscode pytorch环境搭建
时间: 2023-11-09 19:02:42 浏览: 157
在 VS Code 中搭建 PyTorch 环境需要进行以下步骤:
1. 安装 Anaconda 或 Miniconda
2. 创建一个新的 conda 环境
3. 在新环境中安装 PyTorch 和 torchvision
4. 在 VS Code 中配置 Python 解释器为新环境
具体步骤如下:
1. 下载并安装 Anaconda 或 Miniconda,下载地址:https://www.anaconda.com/products/individual 或 https://docs.conda.io/en/latest/miniconda.html
2. 打开 Anaconda Prompt 或终端,输入以下命令创建一个新的 conda 环境:
```
conda create --name pytorch_env python=3.8
```
其中,pytorch_env 是环境名称,python=3.8 表示使用 Python 3.8 版本。
3. 激活新环境:
```
conda activate pytorch_env
```
4. 安装 PyTorch 和 torchvision:
```
conda install pytorch torchvision torchaudio -c pytorch
```
5. 在 VS Code 中打开项目文件夹,按下 Ctrl+Shift+P(或者在菜单栏中选择“View”->“Command Palette”),输入“Python: Select Interpreter”,选择刚才创建的新环境。
阅读全文