- 安装NVIDIA驱动
下载地址
点击开始搜索,自动搜索自己电脑显卡的型号,产品型号记得选择带notebooks的;
点击立即下载,安装包下载完,按默认配置安装即可。
- 安装Anaconda
下载地址
可能需要先注册一个账号,安装包下载完,按默认配置安装即可。
- 进入 Anaconda prompt 命令窗口
先使用清华源,必须使用http
打开C:\Users\你的用户名 文件夹下修改.condarc文档,将其中的内容替换为如下代码: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
回到 Anaconda prompt 命令窗口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
// 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
【win11安装Anaconda和GPU PyTorch】win11安装GPU PyTorch
于 2022-01-08 16:25:17 首次发布