PyTorch2.0 环境搭建详细步骤(Nvidia显卡)

Step 1 、查看显卡驱动版本

Step2、下载CUDA 11.7 或者11.8(我自己用的这个)也行,稍后我会贴出来版本匹配对应表

CUDA Toolkit - Free Tools and Training | NVIDIA Developer

Step3、下载CUDNN

cuDNN 9.0.0 Downloads | NVIDIA Developer

Step4、安装anconda(尽量使用这种方式安装python环境,坑会少一点)

https://www.anaconda.com/download

Step5、配置环境变量

Step6、执行命令安装PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

相关推荐
希艾席帝恩4 小时前
数字孪生如何重塑现代制造体系?
大数据·人工智能·数字孪生·数据可视化·数字化转型
浔川python社4 小时前
关于浔川 AI 翻译项目推进建议的公告
人工智能
武汉海翎光电4 小时前
从数据采集到智能决策:船舶传感器的技术跃迁之路
大数据·人工智能
Coding茶水间5 小时前
基于深度学习的肾结石检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·机器学习·计算机视觉
Blossom.1185 小时前
基于多模态大模型的工业质检系统:从AOI到“零样本“缺陷识别的产线实践
运维·人工智能·python·机器学习·自动化·测试用例·知识图谱
美狐美颜sdk5 小时前
什么是美颜SDK?一套成熟直播美颜SDK需要解决哪些工程技术问题?
人工智能·美颜sdk·第三方美颜sdk·视频美颜sdk·人脸美型sdk
无代码专家5 小时前
无代码:打破技术桎梏,重构企业数字化落地新范式
大数据·人工智能·重构
usrcnusrcn5 小时前
告别PoE管理盲区:有人物联网工业交换机如何以智能供电驱动工业未来
大数据·网络·人工智能·物联网·自动化
雍凉明月夜6 小时前
视觉opencv学习笔记Ⅴ-数据增强(1)
人工智能·python·opencv·计算机视觉