linux anaconda配置pytorch环境
时间: 2023-09-23 17:03:35 浏览: 59
以下是在 Linux 系统中使用 Anaconda 配置 PyTorch 环境的步骤:
1. 安装 Anaconda
在 Anaconda 的官网上下载对应 Linux 系统的 Anaconda 安装包,并安装。
2. 创建虚拟环境
在终端中输入以下命令来创建一个名为 “pytorch” 的虚拟环境:
```
conda create --name pytorch
```
3. 激活虚拟环境
输入以下命令来激活名为 “pytorch” 的虚拟环境:
```
conda activate pytorch
```
4. 安装 PyTorch
输入以下命令来安装 PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
其中,`cudatoolkit=11.1` 表示安装 CUDA 11.1 版本,如果你的系统中没有安装 CUDA,则可以去掉 `cudatoolkit=11.1`。
5. 验证 PyTorch 安装成功
在 Python 环境中输入以下命令来验证 PyTorch 是否安装成功:
```
import torch
print(torch.__version__)
```
如果输出对应的 PyTorch 版本号,则表示安装成功。
6. 退出虚拟环境
输入以下命令来退出虚拟环境:
```
conda deactivate
```
完成以上步骤后,你就可以在 Linux 系统中使用 Anaconda 配置 PyTorch 环境了。