检查cuda和pytorch版本是否匹配
目的：在已经安装pytorch和cuda的环境下，检查当前版本下pytorch和cuda版本是否兼容在cmd输入python，运行python输入import torchprint(torch.__version__)print(torch.cuda.is_available())若最终输出为false则表示pytorch和cuda版本不兼容如下图所示：若输出为true则可以正常使用，如下图所示：
·
目的：在已经安装pytorch和cuda的环境下，检查当前版本下pytorch和cuda版本是否兼容
在cmd输入python，运行python
输入
import torch
print(torch.__version__)
print(torch.cuda.is_available())若最终输出为false则表示pytorch和cuda版本不兼容如下图所示：
若输出为true则可以正常使用，如下图所示：
CSDN联合极客时间，共同打造面向开发者的精品内容学习社区，助力成长！
更多推荐
6
0- 0
已为社区贡献1条内容
所有评论(0)