anaconda安装pytorch1.7.1
时间: 2024-04-13 10:23:28 浏览: 169
anaconda环境下Pytorch设置,压缩包,直接复制粘贴即可
安装PyTorch 1.7.1需要使用Ana进行管理和环境配置。以下是安PyTorch 1.7.1的步骤:
1. 首先,确保已经安装了Anaconda。如果没有安装,请前往Anaconda官方网站下载并按照指示进行安装。
2 打开Anaconda Navigator,点击左侧的"Environments"选项卡,然后点击"Create"按钮创建一个新的环境。
3. 在弹出的对话框中,输入环境的名称(例如"pytorch_env"),选择Python版本(建议选择3.7或3.8),然后点击"Create"按钮创建环境。
4. 创建环境后,返回到"Home"选项卡,选择新创建的环境,点击右侧的"Open Terminal"按钮打开终端。
5. 在终端中,运行以下命令来安装PyTorch 1.7.1:
```
conda install pytorch==1.7.1 torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
这个命令会安装PyTorch、torchvision和torchaudio,并且指定了适用于CUDA 10.2的cudatoolkit版本。如果你没有使用CUDA,可以去掉`cudatoolkit=10.2`这部分。
6. 安装完成后,可以通过运行以下命令来验证PyTorch是否成功安装:
```
python -c "import torch; print(torch.__version__)"
```
如果输出的版本号是1.7.1,则表示PyTorch安装成功。
阅读全文