cupy无法使用,torch.cuda.is_available()

在最近跑代码时,我使用conda新建了一个环境,就从头安装了框架pytorch。但是怎么也跑不通,告诉我cuda版本有问题。我差点为此装一个cuda10,如果不是因为服务器不能被我瞎搞。

决定换一个代码试试时,终于跑通了。
跑通的原因是:这个代码不需要gpu就可以??当我把使用GPU的参数设置为True时,依旧GPU利用率为0??

于是尝试了 torch.cuda.is_available()发现结果为False???

正常环境的情况:
cupy无法使用,torch.cuda.is_available()_第1张图片一般出现这个的原因是因为驱动没有安装好的缘故。解决的方案是:安装nvidia相应的驱动版本和对应的cuda和cudnn。??但是其他环境是正常的

没有安装:cudatoolkit?

去到https://pytorch.org官网,

我们可以选择对应版本的torch进行安装。

cupy无法使用,torch.cuda.is_available()_第2张图片
将会安装很多需要的东西:
cupy无法使用,torch.cuda.is_available()_第3张图片

这个样子我们的服务器才有服务器的亚子~
cupy无法使用,torch.cuda.is_available()_第4张图片

你可能感兴趣的:(python)