解决pytorch报错:Torch not compiled with CUDA enabled

在进行AI深度学习的时候,必不可少需要用到CUDA,但是能正常配置好环境是最难的部分

import torch
print(torch.__version__)
print(torch.cuda.is_available())

在执行后,如果显示false,则代表并没有正确运行
需要找到自己CUDA的版本(自行百度检测方法)
重点是如何下载安装对应的版本
很多博客上都是使用类似于自己找对应的版本手动下载
如下图:
解决pytorch报错:Torch not compiled with CUDA enabled_第1张图片
但是直接在官网选择和自己电脑对应的版本可以一键复制下载命令
自动下载安装
官网传送→
如下如:解决pytorch报错:Torch not compiled with CUDA enabled_第2张图片
选择自己对应的系统,cuda版本即可复制下面的下载命令,直接一键下载
over~

你可能感兴趣的:(python,区块链,分布式,stm32)