1.更新显卡驱动此处跳转

打开cmd输入

nvidia-smi

得以得到显卡相关信息。

2.Anaconda安装

3.pytorch安装

电磁

在Anaconda prompt中输入

conda env list

可以查看环境

键入如下命令

conda create -n pytorch python=3.9 #创建一个pytorch环境

conda activate pytorch  #激活该环境

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

前往官网安装对应版本的pytorch

注意:-c pytorch不要复制