PyTorch安装笔记
Win11+WSL2+Pytorch(GPU版本)
首先在Win11通过微软应用商店安装WSL2 Ubuntu22.0。
主线是参考@七十七的德 的wsl2安装深度学习环境配置这个视频,up写了博客更为详细。但博客网页很难刷新出来,虽然不用FQ。该up是在虚拟环境中安装torch。
我还参考了@Quatttschen 的博客,把torch用pip方法直接安装。虚拟环境的有需要再安装。只安装了CUDA Toolkit 11.8,没安装cudnn,直接安装了torch。 2023.6.23
步骤1:安装anaconda
安装成功后,创建虚拟环境cu118py310:
步骤2:安装Cuda Toolkit
使用nvcc -V能查询出cuda版本11.8,证明安装成功。
步骤3:安装torch
生产环境安装pip install torch:
pip list命令查询结果:
torch$\qquad\qquad$2.0.1+cu118
torchaudio$\qquad$2.0.2+cu118
torchvision$\qquad$0.15.2+cu118
conda list命令查询结果:
torch 2.0.1+cu118 pypi_0 pypi
torchaudio 2.0.2+cu118 pypi_0 pypi
torchvision 0.15.2+cu118 pypi_0 pypi
步骤4:安装Pycharm并配置解释器
在Pycharm2021(专业版)中配置WSL中的python解释器。
选择1:这是WSL生产环境中的解释器:
选择2:这是WSL中虚拟环境的解释器:
Pycharm调用WSL2生产环境torch成功:
Win10+Conda创建虚拟环境+Pytorch(CPU版本)
安装完成后,import torch提示错误。再安装visual studio即可(此处为2022 Community版本)