博文

目前显示的是 三月, 2025的博文

wsl安装cuda

  你有 两种选择 来解决这个问题: 安装与 PyTorch 兼容的 CUDA 版本(推荐) 重新安装适配 CUDA 12.8 的 PyTorch 方案 1:安装 CUDA 11.6(推荐) 既然 PyTorch 是用 CUDA 11.6 编译的,你可以降级 CUDA 版本以匹配 PyTorch。 1. 卸载当前的 CUDA 首先,卸载 CUDA 12.8: sudo apt remove --purge cuda sudo apt autoremove 然后,删除残留的 CUDA 目录: sudo rm -rf /usr/local/cuda* 2. 安装 CUDA 11.6 使用 NVIDIA 官方仓库安装 CUDA 11.6: wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-11-6-keyring_1.0-1_all.deb sudo dpkg -i cuda-11-6-keyring_1.0-1_all.deb sudo apt update sudo apt install -y cuda-11-6 3. 配置环境变量 export CUDA_HOME = /usr/local/cuda-11.6 export PATH = $CUDA_HOME /bin: $PATH export LD_LIBRARY_PATH = $CUDA_HOME /lib64: $LD_LIBRARY_PATH 让配置永久生效: echo 'export CUDA_HOME=/usr/local/cuda-11.6' >> ~/.bashrc echo 'export PATH=$CUDA_HOME/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc 4. 验证安装 nvcc --version 输出应该包含 release 11.6 。 然后重新...