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()看工具怎么用(说明书)

posted @   xjl_ultrasound  阅读(44)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具
点击右上角即可分享
微信分享提示