使用Pytorch查看CUDA和CUDNN
(需要注意涉及到下划线很多由两个下划线构成)
查看pytorch版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
查看torchvision版本
import torchvision
print(torchvision.__version__)
输出

安装opencv
pip install opencv-python
查看python中opencv版本
import cv2
cv2.__version__
输出

查看TensorFlow版本
import tensorflow as tf
print(tf.__version__)
输出
python忽略warning警告
import warnings
warnings.filterwarnings("ignore")
命令行忽略错误的输出
python -W ignore yourscript.py
本文介绍了如何使用Python检查PyTorch、torchvision、OpenCV及TensorFlow等深度学习框架的版本信息,并展示了如何安装和验证这些软件包的安装情况。
1万+

被折叠的 条评论
为什么被折叠?



