PyTorch学习一(环境配置与安装)
PyTorch学习一(环境配置与安装)
anaconda安装
anaconda一个集成的工具包
安装地址:https://repo.anaconda.com/archive/
选择Anaconda3-5.2.0-Windows-x86_64.exe版本(631.1M)
显卡作用:训练加速
显卡配置: 驱动 CUDA Toolkit
在任务管理器检查驱动安装:在性能部分查看GPU如果能正常显示型号则正常
小tips 如何管理环境
(conda create -n pytorch python=3.6)
他就像一个小房子在里面安装环境
使用conda 创建一个环境
(安装时全部y就行)
使用conda activate pytorch来激活
执行成功后可以看到左边括号里的名字变了,变成了新的我们激活的房间
使用pip list命令列出要使用的工具包
安装pytorch
官网地址:https://pytorch.org/
然后查看GPU型号
NVIDIA GeForce RTX 3070 Ti Laptop GPU
安装指令:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
使用nvidia-smi命令查看驱动版本
不够就去官网升级(https://www.nvidia.com/Download/index.aspx?lang=cn)
检测安装
输入python
输入import torch无报错则正常
输入torch.cuda.is_available()
返回true则正常(说明该计算机的GPU可以访问pytorch)
- 感谢你赐予我前进的力量
赞赏者名单
因为你们的支持让我意识到写文章的价值🙏
本文是原创文章,采用 CC BY-NC-ND 4.0 协议,完整转载请注明来自 Sdegree
评论
匿名评论
隐私政策
你无需删除空行,直接评论以获取最佳展示效果