pytorch无法使用cuda

python 复制代码
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号

#False
#0
#None

表明安装失败!查看安装包:

python 复制代码
conda list

发现pytorch的注释为cpu_py310h5e1f01c_1,我安装的不是cuda版本吗,好像是清华源没有pytorch对应的cuda,所以一直给你下载的都是cpu版本。

解决办法:

将conda下载源换回为默认源或者修改默认源为科大源:

python 复制代码
conda config --remove-key channels

切换之后,清华源将不再起作用,起作用的为默认源。

使用下载命令

python 复制代码
pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

参考内容:pytorch无法使用cuda

相关推荐
没有不重的名么1 小时前
Pytorch深度学习快速入门教程
人工智能·pytorch·深度学习
AI-Ming13 小时前
程序员转行学习 AI 大模型: 踩坑记录,HuggingFace镜像设置未生效
人工智能·pytorch·python·gpt·深度学习·学习·agi
剑穗挂着新流苏31213 小时前
205_深度学习的非线性魔法:多层感知机(MLP)与激活函数全解析
人工智能·pytorch·python·深度学习
剑穗挂着新流苏31216 小时前
204_从回归到分类:Softmax 回归、损失函数与多分类实战
人工智能·pytorch·python·深度学习
独隅1 天前
PyTorch 模型部署的 Docker 配置与性能调优深入指南
人工智能·pytorch·docker
靴子学长1 天前
Decoder only 架构下 - KV cache 的理解
pytorch·深度学习·算法·大模型·kv
独隅1 天前
PyTorch 模型性能优化:图像分类与 NLP 模型实战指南
pytorch·性能优化·分类
Narrastory1 天前
明日香 - Pytorch 快速入门保姆级教程(九)
人工智能·pytorch·深度学习
AI视觉网奇1 天前
vllm 踩坑记录 算力匹配
pytorch·python·深度学习
剑穗挂着新流苏3121 天前
202_深度学习的动力源泉:矩阵微积分与自动求导 (Autograd)
人工智能·pytorch·python·深度学习·神经网络