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

相关推荐
白小筠18 小时前
Pytorch之张量的基本操作
人工智能·pytorch·python
热爱生活的猴子19 小时前
PyTorch导出ONNX报错(ShapeInferenceError)问题笔记(含dynamo=False作用解析)
人工智能·pytorch·笔记
YuanDaima204820 小时前
解决Conda环境下RTX 50系列显卡PyTorch+Transformers+PEFT微调报错
人工智能·pytorch·笔记·python·深度学习·机器学习·conda
郝学胜-神的一滴21 小时前
张量维度操控心法:从reshape到升维降维,吃透PyTorch形状操作的底层逻辑
人工智能·pytorch·python·深度学习·程序人生·算法·机器学习
罗罗攀1 天前
PyTorch学习笔记|单层神经网络
人工智能·pytorch·笔记·神经网络·学习
纤纡.2 天前
基于 PyTorch 手动实现 CBOW 词向量训练详解
人工智能·pytorch·python·深度学习
ZPC82102 天前
MoveIt Servo 控制真实机械臂
人工智能·pytorch·算法·性能优化·机器人
Web极客码2 天前
PyTorch 实战:为神经网络开启“实时自愈”模式
人工智能·pytorch·神经网络
weiwei228443 天前
torch.nn神经网络详解
pytorch·mnist·手写识别
rebekk3 天前
PyTorch Dispatcher介绍
人工智能·pytorch·python