本文指导如何在Windows 11的WSL2（Ubuntu 20.04）环境中，安装CUDA 11.4并从源代码编译PyTorch，以利用GPU加速。首先升级到Windows 11，然后在WSL2中安装CUDA，接着在Windows 11上安装NVIDIA预览驱动。之后在WSL2中安装Anaconda并创建Python环境。最后，按照PyTorch官方指南编译源代码，并测试PyTorch是否成功使用GPU加速。
For detailed information on step 1 ~ step 4, please refer to CUDA on WSL :: CUDA Toolkit Documentation (nvidia.com)
https://docs.nvidia.com/cuda/wsl-user-guide/
- Upgrade to Windows11
- install cuda on WSL2(in my case, I am using Ubuntu 20.04 and cuda 11.4)
- install Nividia preview drivers on windows11
- install anaconda on WSL2
- create env in anaconda(in my case, I am using python=3.8.11)
- follow "build from source" steps on
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
