windows10 配置pytorch环境

windows10 配置pytorch环境

前不久有台实验室电脑重装，需要重新配一下环境，写篇博客记录一下

安装Annaconda

只能说懂得都懂，annaconda管理python环境确实舒服。首先安装一手Annaconda，我习惯性去清华的镜像源下载，比直接官网下快的多了。
安装完成后，打开控制台创建一个环境，输入如下命令，创建一个环境

conda create -n your_env_name python=3.9

创建完成后，输入

conda activate your_env_name

切换到你的环境下

安装Pytorch

首先看看电脑的cuda版本（后面需要用到）
在控制台输入

nvidia-smi

如图我的cuda版本为11.7。

得到cuda版本后就比较简单了（如果你的电脑没有安装cuda你就需要去官网下载你显卡对应的cuda）。前往pytorch官网选择对应的cuda版本。

在控制台输入生成的命令，完成安装
验证一下：
在控制台输入python，再输入import torch
如果没有报错就成功了。