PyTorch 安装

本文基于conda安装,请确保已经安装好Anaconda,可参考上一篇文章安装Anaconda。

一、确定本机CUDA版本

基于N卡,基于N卡,基于N卡

nvidia-smi  #N卡用此命名查看,N卡如果没有此命令,先去更新显卡驱动

PyTorch 安装_第1张图片

二、安装

前往官网安装适合自己的版本。确保你当前的版本大于等于你选择的版本即可,向下兼容
官网地址:https://pytorch.org/get-started/locally/
PyTorch 安装_第2张图片

三、复制命令进conda执行

先激活CONDA的环境,然后输入

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

记得确认输个Y,进入以下界面后就可以慢慢等了。
PyTorch 安装_第3张图片

你可能感兴趣的:(python,pytorch,人工智能,python)