在WSL2下配置Pytorch(Linux+Anaconda+PyTorch-GPU)
在WSL2下配置Pytorch环境,首先确保你的Windows系统支持GPU加速,并且已经安装了WSL2以及适用于Linux的Windows子系统。以下是配置步骤:
安装Anaconda:
下载Anaconda安装脚本并安装,可以从Anaconda官网获取安装脚本链接。
创建一个新的conda环境并激活它:
conda create -n pytorch-gpu python=3.8 conda activate pytorch-gpu
安装PyTorch:
访问PyTorch官网的安装指南(https://pytorch.org/get-started/locally/),使用conda命令安装适合你系统的CUDA版本的PyTorch。例如,如果你的CUDA版本是11.1,可以使用以下命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
请确保替换cudatoolkit=11.1
为你的Windows支持的版本。
注意:如果你的Windows没有GPU或者不支持CUDA,你可以安装CPU版本的PyTorch:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
以上步骤在WSL2环境中应该可以正常工作,但如果遇到具体的错误,请根据错误信息进行相应的调整。
评论已关闭