ubuntu查看安装的pytorch/cuda版本

2024年4月1日 345次阅读来源: pytorch

使用命令：

user@home:~$ python
Python 3.6.3 |Anaconda custom (64-bit)| (default, Oct 13 2017, 12:02:49)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.0.1.post2'

和

user@home:~$ cat /usr/local/cuda/version.txt
CUDA Version 9.0.176

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/wanghui-garcia/p/10576623.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。