1.cuda是否安装正确

nvcc -V

 

 

2.cudnn是否安装正确 

nvidia-smi

3.安装Anaconda

4.更换清华源-不做描述

5.安装虚拟环境

6.安装pytorch

https://pytorch.org/get-started/locally/

7.出现问题排查:

 

我找了很久,前面的步骤都是正确的,但是一直都是安装不上torch,最后发现一直安装的都是cpu版本的,并未安装GPU版本,并且torch.cuda.is_available()一直都是False,所以,对python版本进行排查。发现是我创建的虚拟环境是3.6,所以我安装了一个py3.8虚拟环境并且对此进行重新安装。 安装正确

 

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐