问题:在编译cuda代码时需要安装pytorch-cuda,编译过程报错
解法:
下载gcc
gcc指定版本,自行编译安装
在这里下载想要版本的gcc
wget http://ftp.gnu.org/gnu/gcc/gcc-11.3.0/gcc-11.3.0.tar.gz
解压
tar -xvf gcc-11.3.0.tar.gz
编译gcc
cd gcc-11.3
./contrib/download_prerequisites
mkdir build
cd build
指定安装目录
…/configure --prefix=/mypath/gcc-11.3 --disable-multilib # --prefix指定安装目录
make -j250
make install
添加环境变量
vim ~/.bashrc
在 .bashrc 中改变环境变量,在最后添加:
export PATH="/mypath/gcc-11.3/bin:$PATH"
export LD_LIBRARY_PATH="/mypath/gcc-11.3/lib:$LD_LIBRARY_PATH"
export LD_LIBRARY_PATH="/mypath/gcc-11.3/lib64:$LD_LIBRARY_PATH"
启动bashrc
source ~/.bashrc
which gcc # 查看是否是正确的path
gcc --version # 查看版本
编译代码时cuda报错,查询后得知cudatoolkit与gcc版本不匹配
cudatoolkit需要用大于等于11.6.2版本
cuda下载地址:cudatoolkit
自己编译安装。