Anaconda 是数据科学和机器学习领域非常核心的工具集,可以理解为一个专门为Python/R语言打造的“一站式开发环境管理平台”。对于AI开发者来说,它几乎是标配。

用conda安装虚拟环境 ,包括cuda, tooltip ,请参考如下步骤 , 以下都是以命令行的方式安装,win10系统可以用PowerShell
本文安装适用于windows系统,linux系统

1、创建环境

以下命令创建名字为myEnv的虚拟环境,python版本选择 3.10

conda create --name myEnv python=3.10

2、激活环境

conda activate myEnv

3、安装环境

conda 对应依赖 pytorch cuda 安装conda 安装对应依赖版本,请参考以下文档,选择合适的版本,我们一般选择 conda 来安装非pip,根据经验用pip命令失败率高一些
https://pytorch.org/get-started/previous-versions/
Linux和Windows 系统安装命令是一样的,例如安装pytorch 2.5.1 cuda 11.8可以选择以下命令

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1  pytorch-cuda=11.8 -c pytorch -c nvidia

安装之前,可以先换源,添加国内镜像源,这样下载安装会快很多

安装前检查环境需要几分钟,安装过程一般需要20分钟左右
win10安装可能到后面会卡住,直接按Enter就行

4、nvccHome 安装

以上3个步骤安装完成 ,还需要安装 nvcc 环境,有的模型要检查,比如我安装11.8 版本命令如下

conda install cuda-nvcc=11.8 -c nvidia

5、验证cuda , pytorch 安装结果 ,命令输入以下代码

python
import torch
torch.cuda.is_available()
print(torch.__version__)

如果is_available输出为true,则pytorch安装成功能识别显卡

验证安装环境识别cuda

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐