1 查看电脑对应cuda版本

  1. 【win-R】=> 【cmd】=> 确定
    在这里插入图片描述
  2. 执行nvidia-smi,可以从图中看出,该电脑的CUDA Version为:11.6
    在这里插入图片描述

2 官网查找对应的版本的安装语句

PyTorch官网:https://pytorch.org/

2.1 可以直观的选择对应的版本

如果没有在以下版本中,请点击下方的Previous versions of PyTorch
在这里插入图片描述

2.2 Previous versions of PyTorch

按下图的说明,选取合适的pytorch版本

在这里插入图片描述

3 Anaconda中使用命令行安装pytorch

Anaconda Prompt 中,打开对应的环境(如下图,打开了forall),安装pytorch
在这里插入图片描述

# 打开forall环境
conda activate forall
# 执行命令,安装pytorch,torchvision,torchaudio,pytorch-cuda
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

等待时间可能会比较久,不过,值得等候。

Logo

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

更多推荐