- 安装NVIDIA驱动
下载地址
点击开始搜索,自动搜索自己电脑显卡的型号,产品型号记得选择带notebooks的;
点击立即下载,安装包下载完,按默认配置安装即可。
安装Anaconda
下载地址
可能需要先注册一个账号,安装包下载完,按默认配置安装即可。
进入 Anaconda prompt 命令窗口
先使用清华源,必须使用http
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --set show_channel_urls yes打开C:Users你的用户名 文件夹下修改.condarc文档,将其中的内容替换为如下代码:
channels: - defaults show_channel_urls: true default_channels: - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 custom_channels: conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud回到 Anaconda prompt 命令窗口
// PyTorch是名字,可以改,必须指定python版本号 conda create -n PyTorch python=3.9 // 查看 Anaconda 的所有环境 conda info --envs进入pytorch官网
官网地址
选择相应配置,复制配置下面的命令回到 Anaconda prompt 命令窗口
输入下面的命令进入PyTorch环境
conda activate PyTorch
输入刚刚复制的命令,等待下载完成
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch测试安装成功
重新进入 Anaconda prompt 命令窗口
// 进入python环境 python // 测试 import torch torch.cuda.is_available()输出为
True至此安装成功



