最近重新装了下系统,然后初始化了cuda环境,记录下这几条命令
1,判断pytorch是否能加速
print torch.cuda.is_available()
print(torch.backends.cudnn.version())
3,tensorflow是否支持GPU
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
4,torch中的cuda版本
torch.version.cuda
最近重新装了下系统,然后初始化了cuda环境,记录下这几条命令
1,判断pytorch是否能加速
print torch.cuda.is_available()
print(torch.backends.cudnn.version())
3,tensorflow是否支持GPU
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
4,torch中的cuda版本
torch.version.cuda