linux下查看torch版本和cuda版本

python3和python2一样
可以在控制台进行输入

>>import torch
>>print(torch.__version__)

终端中写入

cat  /usr/local/cuda/version.txt

回车

posted @ 2021-01-31 15:15  小Aer  阅读(2448)  评论(0)  编辑  收藏  举报