由于linux系统可以同时安装多个版本的cuda,我们有时候无法知道pytorch之类的深度学习框架基于哪一个版本,直接查看系统cuda版本有以下两种办法,但会出现问题:
命令行输入 cat /usr/local/cuda/version.txt 回车得到10.1版本,
或者 命令行输入 nvcc -V 回车得到10.0版本 …
这时候直接在命令行下输入下面的语句查看才准确
python
import torch
print(torch.version.cuda)
由于linux系统可以同时安装多个版本的cuda,我们有时候无法知道pytorch之类的深度学习框架基于哪一个版本,直接查看系统cuda版本有以下两种办法,但会出现问题:
命令行输入 cat /usr/local/cuda/version.txt 回车得到10.1版本,
或者 命令行输入 nvcc -V 回车得到10.0版本 …
这时候直接在命令行下输入下面的语句查看才准确
python
import torch
print(torch.version.cuda)