查看cuda版本和torch版本

前言

记录查看cuda版本、torch版本的代码、conda常见命令和踩坑心得

二、使用步骤

1.查看torch版本：

python
import torch
print(torch.version)
exit()

2.查看cuda版本：

nvcc -V

上图是实际安装的cuda版本。

nvidia-smi

上图是最高支持的cuda版本

---

3.查看whl适配版本

pip debug --verbose

上图是python适配的whl版本

4.如果系统有多个cuda，设置系统cuda

vim ~/.bashrc

更新

source ~/.bashrc

5.新建conda环境并指定python版本

conda create -n yourname python=3.8.2

6.查看系统中存在的conda环境

conda env list

7.删除系统中存在的conda环境

conda env remove --name <yourname>

8.遇到torch-scatter torch-sparse torch-cluster torch-spline-conv 这几个包的问题，通常是和torch版本不匹配造成的。解决方案：在 https://pytorch-geometric.com/whl/ 里面找自己torch版本对应的包即可。