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

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>