【无标题】

Ubuntu 相关 小记

cuda版本相关：

服务器中可以更换cuda版本，一些相关命令记录如下：

1. pytorch 中验证cuda安装成功

python
import torch
torch.cuda.is_available()

torch 查看 cuda 版本

import torch.utils
import torch.utils.cpp_extension
torch.utils.cpp_extension.CUDA_HOME
torch.version.cuda

ubuntu 系统中的cuda版本应该尽量与 conda 中安装的 cudatoolkit保持一致
系统中的可以安装多个cuda版本
切换的方法在环境变量里面

export PATH="/usr/local/cuda-11.0/bin:$PATH"
export CUDA_HOME="/usr/local/cuda-11.0/"
export LD_LIBARAY_PATH="/usr/local/cuda-11.0/lib64:/usr/local/cuda-11.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH"

当cuda加入环境变量成功以后，nvcc -V应该可以输出版本号

查看环境变量值的方法：

echo $PATH

gcc g++版本相关：

1.服务器中同样可以安装不同版本的 gcc g++
2.查看安装版本的方法
跳至环境变量下/bin 或者 /usr/bin下

ls gcc*
ls g++*

版本切换方法：
可以在安装编译前使用命令：

export CXX=g++-7
python setup.py install

注意看清自己在环境路径中搜索出来的gcc g++的表示形式。有没有 -