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

相关推荐
风虎云龙科研服务器2 分钟前
从芯片互连到机器人革命:英伟达双线出击,NVLink开放生态+GR00T模型定义AI计算新时代
人工智能·机器人
qq_436962186 分钟前
奥威BI:打破AI数据分析伪场景,赋能企业真实决策价值
人工智能·数据挖掘·数据分析·ai数据分析
Christo312 分钟前
LNCS-2009《Adaptive Sampling for $k$-Means Clustering》
人工智能·算法·机器学习·kmeans
AI.NET 极客圈32 分钟前
AI与.NET技术实操系列(八):使用Catalyst进行自然语言处理
人工智能·自然语言处理·.net
AI.NET 极客圈33 分钟前
AI与.NET技术实操系列(七):使用Emgu CV进行计算机视觉操作
人工智能·计算机视觉·.net
Cloud Traveler1 小时前
生成式 AI:解锁人类创造力的智能引擎
人工智能
正在走向自律1 小时前
探索Dify:开启大语言模型应用开发新时代
人工智能·语言模型·自然语言处理·llm·didy
MARS_AI_1 小时前
云蝠语音智能体——电话面试中的智能助手
人工智能·自然语言处理·面试·职场和发展·交互·信息与通信
whaosoft-1431 小时前
w~大模型~合集4
人工智能
Mr.小海1 小时前
AI 商业化部署中,ollama 和 vllm 的选型对比
人工智能·大模型