Ubuntu 相关 小记
cuda版本相关:
服务器中可以更换cuda版本,一些相关命令记录如下:
- 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++的表示形式。有没有 -