如何在 Windows 上通过 Anaconda 安装 PyTorch 及 CUDA 环境

当我们需要使用 PyTorch 进行深度学习任务时,需要在本地计算机上安装 PyTorch 及相应的 CUDA 环境。本文将介绍如何在 Windows 操作系统上通过 Anaconda 安装 PyTorch 及 CUDA 环境。

首先,我们需要下载并安装 Anaconda。可以从官网下载安装程序,并选择合适的版本进行安装。安装过程中建议将 Anaconda 安装在 C 盘中。

安装完成后,我们可以在开始菜单中找到 Anaconda Prompt,并打开该命令行工具。在命令行中,我们可以使用 conda 命令创建一个新的虚拟环境,并在环境中安装 PyTorch 和 CUDA。

具体来说,我们可以使用以下命令创建一个名为 pytorch-g 的新环境,并在环境中安装 Python 3.6:

conda create -n pytorch-g python=3.6

然后,我们可以使用以下命令激活该环境:

conda activate pytorch-g

接着,我们可以使用 pip 命令检查环境中的包是否正确:

pip list

在环境激活后,我们需要下载并安装 PyTorch 和 CUDA。我们可以在 PyTorch 的官网上找到相应的安装程序,但在 Windows 操作系统上,建议使用 conda 命令进行安装。以下是安装命令:

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

其中,pytorch-cuda=11.6 表示安装的 PyTorch 版本支持 CUDA 11.6 环境。安装过程可能需要一些时间,请耐心等待。

安装完成后,我们可以使用以下命令检查环境中的包是否正确:

conda list

如果环境中已经安装了 PyTorch 和 CUDA 相关的包,那么我们可以在 Python 环境中导入 PyTorch 并检查 CUDA 是否可用:

python
import torch
torch.cuda.is_available()

如果返回 True,则说明 PyTorch 和 CUDA 环境已经正确安装。

给TA打赏
共{{data.count}}人
人已打赏
站长笔记

关于国内conda安装cuda11.6+pytorch的那些事。

2023-2-20 19:39:26

站长笔记

Windows&Linux下一键切换pip源

2023-2-26 0:07:46

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索