前言

记录查看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版本对应的包即可。

9 遇到Modules找不到的问题时，若module是自己代码文件里面的内容可以通过将文件路径加入系统的环境变量中解决。

import sys
sys.path.append('/home/user/python/')