Torch和cuda版本问题记录

学校的Linux服务器

做了一个Streamlit的小网站，想把一个torch的去噪任务集成上去。

streamlit的环境是py3.8，去噪任务环境是py3.7，torch版本是1.13.1

---

查了一下torch1.1好像支持到3.7，直接把3.7拿到3.8上会不会报错，果然报错

根据以往的经验，环境问题嘛肯定是，我又改了torch的版本，然后又出来cuda的错误了

会不会是这个cuda和这个torch的问题

我3.7+torch1131不跑的挺好的嘛cuda我又没动过，怎么回事？

百度：如何查看cuda版本？ 答：nvcc -V

啊？他说我没有，那这是什么，我以前怎么用的显卡？

---

又百度: nvcc -V没反应，但是能用gpu

答：conda或者pip安装torch的时候自带了一个cuda，如下

import torch
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
print(torch.cuda.nccl.version())

True
1.7.1+cu110
11.0
8005
2708

我们可以直接使用torch自带的cuda

安装torch的时候不仅可以指定torch的版本还可以指定torch自带的cuda的版本

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch