在进行AI深度学习的时候,必不可少需要用到CUDA,但是能正常配置好环境是最难的部分
import torch
print(torch.__version__)
print(torch.cuda.is_available())
在执行后,如果显示false,则代表并没有正确运行
需要找到自己CUDA的版本(自行百度检测方法)
重点是如何下载安装对应的版本
很多博客上都是使用类似于自己找对应的版本手动下载
如下图:
但是直接在官网选择和自己电脑对应的版本可以一键复制下载命令
自动下载安装
官网传送→
如下如:
选择自己对应的系统,cuda版本即可复制下面的下载命令,直接一键下载
over~