Anaconda安装GPU版本pytorch
发布时间
阅读量：
阅读量
很多问题被重复提出（许多问题被重复提出），然而这些解决方案往往过于繁琐（然而这些解决方案往往过于繁琐），使得即使是完全没有编程基础的新手也感到困惑（使得即使是完全没有编程基础的新手也感到困惑）。实际上安装PyTorch非常简单：只需一行代码即可从官方文档完成安装（实际上安装PyTorch非常简单：只需一行代码即可从官方文档完成安装）。
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
AI写代码为了方便新手快速搭建深度学习环境，在安装PyTorch相关依赖库时需要注意一些细节问题
conda install PyTorch PyTorch Vision TorchAudio PyTorch CUDA=11_8 -c PyTorch -c NVIDIA
该指令的功能是使用conda执行下载操作以获取pytorch torch torchvision以及torchaudio这几个软件包。其中参数-c pytorch和-c nvidia分别是从官方pytorch和nvidia指定路径处获取软件包。然而由于官方路径pytorch在国内网络访问速度较慢为此建议采用清华大学镜像库作为替代资源。
conda install pytorch torchvision torchaudio -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
AI写代码替代；通过这个路径下载pytorch(torch)、torchvision和torchaudio这三个库包。如果仅运行该指令进行安装，则会使得安装的pytorch为CPU版本。
从cuda 11.6起采用pytorch-cuda取代cudatoolkit，并规定仅限于nvidia官方渠道获取
conda install pytorch-cuda=11.8 -c nvidia
AI写代码故合并两条语句一起安装的指令应该是
conda install pytorch torchvision torchaudio pytorch-cuda=11.8(改为相应版本） -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ -c nvidia
AI写代码这样安装的pytorch就是GPU版本。
全部评论 (0)
还没有任何评论哟~