Pytorch笔记|小土堆|P1-5|Pytorch环境配置
Pytorch环境安装及配置
1、创建conda环境,名为pytorch
conda create -n pytorch python=3.10
2、在任务管理器的性能中确认显卡,是否支持CUDA。其次,确认显卡驱动,cuda9.2支持396.26以上的驱动,可以在命令行输入nvidia-smi
来看自己驱动是否满足要求,如果低于396.26,可以使用各种管家更新驱动或者去官网下载最新驱动,https://www.nvidia.com/Download/index.aspx?lang=cn
3、去Pytorch官网根据自己的需求获取安装代码
*没有能用CUDA的英伟达显卡(去CUDA官网确认)则选CPU
4、命令行输入conda activate pytorch
以进入之前创建的pytorch环境,运行从官网复制过来的安装代码
*这一步如果很慢或一直失败,可参考的解决方法:先添加清华源,并把复制来的安装代码最后的-c pytorch去掉后运行,亲测会安装得很快(建议关掉vpn)。具体见https://blog.csdn.net/zzq060143/article/details/88042075
5、安装后用pip list
确认已安装的package,可以看到torch在列表里。另一个方法是命令行输入python
,在python环境下输入import torch
,不报错说明安装成功。如果安装了CUDA版本,继续输入torch.cuda.is_available()
,返回True则成功
Python编辑器
Pycharm:新建项目,选existing intepreter,找到之前创建的名为pytorch的环境。
在之前创建的pytorch环境下安装jupyter notebook:
conda install jupyter notebook
或者
conda install nb_conda_kernels
然后命令行输入jupyter notebook
跳转网页
Package学习
dir()
看package里有哪些分区和工具
help()
看工具怎么用(说明书)
本文来自博客园,作者:xjl-ultrasound,转载请注明原文链接:https://www.cnblogs.com/xjl-ultrasound/p/18336776
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具