安装CUDA以及GPU版本的pytorch

使用pytorch进行深度学习的时候,往往想用GPU进行运算来提高速度。于是搜索便知道了CUDA。

下面给出一个自检的建议:

  1. 检查cuda的版本是否适配自己的GPU。

打开NVDIA控制面板,点击左下角"系统信息",然后就可以看到NVDIA GPU的详细信息,其中就包含了CUDA的版本。在官网安装合适版本的cuda-toolkit。

  1. 安装了cuda,但是命令行输入nvcc -V 报错显示没有nvcc

这时候可能没有将CUDA添加到环境变量。检查系统变量中是否包含了CUDA_PATH ,以及CUDA_PATH_Vx.x , 以及PATH中是否包含了cuda的bin目录。

  1. 在命令行输入nvcc -V 可以正常运行,但是在python中使用print(torch.cuda.is_available()) 显示的是False。 这时候考虑torch安装错误。

(直接使用pip install torch 会安装cpu版本的torch,导致无法使用cuda)

卸载原先的torch,pip uninstall torch

然后在PyTorch官网找到正确版本进行下载安装。我的CUDA版本是12.6,PyTorch官网的最新版本支持CUDA12.4 安装命令如下:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

此时在python中再运行print(torch.cuda.is_available())就会显示True了。

相关推荐
说私域2 分钟前
基于开源AI大模型、AI智能名片与S2B2C商城小程序源码的用户价值引导与核心用户沉淀策略研究
人工智能·开源
亲持红叶3 分钟前
GLU 变种:ReGLU 、 GEGLU 、 SwiGLU
人工智能·深度学习·神经网络·激活函数
说私域3 分钟前
线上协同办公时代:以开源AI大模型等工具培养网感,拥抱职业变革
人工智能·开源
群联云防护小杜5 分钟前
深度隐匿源IP:高防+群联AI云防护防绕过实战
运维·服务器·前端·网络·人工智能·网络协议·tcp/ip
惜.己8 分钟前
使用python读取json数据,简单的处理成元组数组
开发语言·python·测试工具·json
摘星编程10 分钟前
构建智能客服Agent:从需求分析到生产部署
人工智能·需求分析·智能客服·agent开发·生产部署
不爱学习的YY酱13 分钟前
信息检索革命:Perplexica+cpolar打造你的专属智能搜索中枢
人工智能
whaosoft-1431 小时前
51c自动驾驶~合集7
人工智能
都叫我大帅哥1 小时前
Python的Optional:让你的代码优雅处理“空值”危机
python
曾几何时`3 小时前
基于python和neo4j构建知识图谱医药问答系统
python·知识图谱·neo4j