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

相关推荐
知行力3 小时前
【GitHub每日速递 20251111】PyTorch:GPU加速、动态网络,深度学习平台的不二之选!
pytorch·深度学习·github
love530love6 小时前
【笔记】xFormers版本与PyTorch、CUDA对应关系及正确安装方法详解
人工智能·pytorch·windows·笔记·python·深度学习·xformers
为你写首诗ge7 小时前
【python】python安装使用pytorch库环境配置
pytorch·python
Danceful_YJ7 小时前
30.注意力汇聚:Nadaraya-Watson 核回归
pytorch·python·深度学习
wa的一声哭了11 小时前
hf中transformers库中generate的greedy_search
android·java·javascript·pytorch·深度学习·语言模型·transformer
Danceful_YJ11 小时前
36.优化方法
人工智能·pytorch·python·深度学习·优化器算法
大连好光景13 小时前
LSTM模型做分类任务2(PyTorch实现)
人工智能·pytorch·lstm
Sanlings14 小时前
ComfyUI+RX5700XT+Ubuntu25.04运行配置
pytorch·ai·comfyui·amd·rocm·rx5700xt·ubuntu25.04
Danceful_YJ1 天前
31.注意力评分函数
pytorch·python·深度学习
CODE_RabbitV1 天前
【1min 速通 -- PyTorch 张量数据类型】张量类型的获取、转化与判别
人工智能·pytorch·python