pytorch在anaconda中安装全过程包括anaconda环境配置
时间: 2024-05-12 16:11:44 浏览: 179
在Anaconda中安装PyTorch需要以下步骤:
1. 首先下载并安装Anaconda,官网下载地址为:https://www.anaconda.com/products/individual
2. 创建新的虚拟环境(可选):打开Anaconda Prompt,输入命令`conda create -n myenv python=3.6`,其中`myenv`是自定义的虚拟环境名称,`python=3.6`是选择Python版本,可以根据需要进行修改。
3. 激活虚拟环境:输入命令`conda activate myenv`,其中`myenv`是上一步创建的虚拟环境名称。
4. 安装PyTorch:在Anaconda Prompt中输入以下命令:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
上述命令会安装PyTorch及其依赖项,并且指定使用CPU版本。如果要安装GPU版本,可以将`cpuonly`改为对应的GPU版本,例如:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c conda-forge
```
其中`cudatoolkit=11.1`是指定CUDA版本号。
5. 安装Jupyter Notebook(可选):在Anaconda Prompt中输入以下命令:
```
conda install jupyter notebook
```
安装完成后,输入命令`jupyter notebook`即可启动Jupyter Notebook。
6. 关闭虚拟环境(可选):在Anaconda Prompt中输入命令`conda deactivate`即可关闭当前虚拟环境。
阅读全文