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
如果没有报错就成功了。