Torch+CUDA安装笔记

WSL2

首先在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成功:

Win11

posted @ 2023-06-23 18:09  萧驭  阅读(223)  评论(0编辑  收藏  举报