这里需要注意环境配置,之前使用本地计算机搭建linux服务器的时候使用的是WSL+Ubuntu,许多人使用默认选项安装的nvcc版本可能过低,比如说11.5,但是目前pytorch官网提供的针对CUDA的版本最低都是11.8,因此在安装nvcc的时候,也要升级到对应11.8以上才可以。
nvcc11.8的安装可以参考链接添加链接描述
根据官网提示,使用下面的命令可以把相关的文件下载好,
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo