pytorch的安装--测试

---------------------------------------------测试是否成功---------------------------------------

python测试程序：

import  torch

#返回当前设备索引

# torch.cuda.current_device()

#返回GPU的数量

# torch.cuda.device_count()

#返回gpu名字，设备索引默认从0开始

# torch.cuda.get_device_name(0)

#cuda是否可用

# torch.cuda.is_available()

# 查看pytorch 版本

# print(torch.__version__)

# 查看pytorch对应的cuda 版本

# print(torch.version.cuda)

# 判断pytorch是否支持GPU加速

# print (torch.cuda.is_available())

# 使用的是GPU的，会输出：cuda:0

# 使用的不是GPU的，会输出：cpu

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

print(device)

上述结果为cuda:0表示pytorch安装成功

至此pytorch安装完毕，比较新的机器的话，安装会比较顺利，安装完毕后即可正常使用

备注：有些时候虽然测试成功，但在实际使用时会提示GPU太老，并且申请内存失败

针对老旧GPU想顺利使用的话，请使用之前安装中查询到的最低版本进行安装。

但是上述都不行的话，劝你放弃，买个新机器吧。

还不想放弃的话，只有下面一条路可行：即源码安装 耗时耗力，问题千奇百怪。很可能几天出去了。

pytorch的安装--源码_huoshi191的博客-优快云博客