WSL 配置CUDA+PyTorch环境

最新推荐文章于 2024-10-08 19:58:04 发布

Gerund

最新推荐文章于 2024-10-08 19:58:04 发布

阅读量436

点赞数

文章标签： cuda windows pytorch

本文链接：https://blog.csdn.net/gerund/article/details/121347529

版权

本文档介绍了如何在Windows子系统Linux（WSL）中配置CUDA 11.1及PyTorch环境，由于官方文档只提供CUDA 11.5的安装步骤，故进行了适配。内容包括安装CUDA，更新pip，安装PyTorch，以及一些杂项操作，如Git Clone特定分支和卸载错误的CUDA版本。

摘要由CSDN通过智能技术生成

Pytorch兼容的最高CUDA是11.1, 而CUDA官方文档关于WSL给出的只有CUDA11.5的安装步骤，所以在此整理了一下

提示:WSL还是有很多不兼容想在自己电脑跑各种linux包还是老老实实装双系统

安装CUDA

根据CUDA11.5官方文档改编：

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.1.0/local_installers/cuda-repo-wsl-ubuntu-11-1-local_11.1.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-1-local_11.1.0-1_amd64.deb
sudo apt-key