欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

Windows+pytorch【配置与试运行】

时间:2023-08-17

1、查看电脑是否配置有nvidia的GPU

【运行】-->输入命令“dxdiag”查看显示驱动即可

2、进入nvidia的官网查找与显卡匹配的驱动以及CUDA并安装官方 GeForce 驱动程序 | NVIDIA下载最新版官方 Geforce 驱动程序,提升 PC 游戏体验和应用程序速度。https://www.nvidia.cn/geforce/drivers/

3、查看是否安装完成

【运行】-->输入“cmd”调出命令行窗口,并在窗口内输入

nvidia-smi

若安装成功,则会出现

 4、查看电脑的python版本

python --version

5、下载python对应版本的miniconda

Miniconda — Conda documentationhttps://docs.conda.io/en/latest/miniconda.html6、下载对应版本的pytorch

Start Locally | PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/

 在anaconda环境里运行上述命令即可

pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

至此,pytorch的环境基本配置完成

可以采用李沐《动手学深度学习》中的附件网络来进行测试

1、进入《动手学深度学习》官网,下载“jupyter”对应的d2l包并解压

《动手学深度学习》 — 动手学深度学习 2.0.0-beta0 documentation

2、安装对应包

pip install jupyter d2l

3、打开jupyter记事本

jupyter notebook

4、找到之前下载好对应目录下的教材资源d2l包,里面pytorch部分自行寻找网络运行即可

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。