Linux查看cuda和pytorch版本，解决版本不匹配问题

写这篇文章主要是方便记录Linux一些常见的命令。

1.查看cuda版本：

nvcc -V

或者

cat /usr/local/cuda/version.txt

2. 查看pytorch版本：

python
import torch
print(torch.__version__)

3. 查看pytorch和cuda版本是否匹配：

print(torch.cuda.is_available())

返回True说明版本是匹配的。

4.解决pytorch和cuda版本不匹配问题：

如果返回False说明版本不匹配，需要到pytorch官网下载cuda版本对应的pytorch

例如：cuda版本为10.2，可通过

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

下载1.8版本的pytorch。