1. 基础环境
- NVIDIA 驱动:安装方式采用命令行或者图形界面(update)
- cuda:bash方式安装
- 验证:nvcc -V || torch.version.cuda 版本一致。
如果nvcc -V版本不对,cat /usr/local/cuda/version.txt 以及 cd /usr/local && stat cuda
再次验证:将bashrc文件中的cuda的8.0(错误版本)改为10.0(对应版本),再source一下
sudo gedit ~/.bashrc # 改为10.0 export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} source ~/.bashrc