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

相关推荐
ydmy7 小时前
注意力机制(个人理解)
pytorch·python·深度学习
AI技术增长10 小时前
Pytorch图像去噪实战(四):Attention UNet图像去噪实战,让模型重点恢复边缘和纹理区域
人工智能·pytorch·python
隔壁大炮11 小时前
Day07-RNN介绍
人工智能·pytorch·rnn·深度学习·神经网络·算法·numpy
AI技术增长12 小时前
Pytorch图像去噪实战(二):用UNet解决DnCNN细节丢失问题(结构解析+完整代码+踩坑总结)
人工智能·pytorch·python
AI技术增长13 小时前
Pytorch图像去噪实战(五):FFDNet可控图像去噪实战,用噪声强度图解决不同噪声等级问题
pytorch·python·深度学习
AI技术增长15 小时前
Pytorch图像去噪实战(三):ResUNet图像去噪模型实战,解决UNet深层训练不稳定问题
人工智能·pytorch·深度学习
ting945200015 小时前
动手学深度学习(PyTorch版)深度详解(1)(含实操+避坑)
pytorch·深度学习·学习
nervermore99015 小时前
3. 人工智能学习-PyTorch框架学习
人工智能·pytorch·学习
数据牧羊人的成长笔记17 小时前
认识深度学习_PyTorch入门+神经网络基础+卷积神经网络+迁移学习+生成对抗网络_GAN+CNN目标检测+循环神经网络与NLP
pytorch·深度学习·神经网络
kishu_iOS&AI17 小时前
NLP —— 文本预处理
人工智能·pytorch·python·自然语言处理