欢迎光临散文网 会员登陆 & 注册

WSL2安装Pytorch

2022-06-15 00:00 作者:狠爷  | 我要投稿

WSL2

管理员运行PowerShell

确认内核版本

升级

重启

安装ubuntu

我比较喜欢Windows Terminal

新的标签可以直接打开ubuntu

首次进入ubuntu需要创建用户名及密码

安装GCC

Nvidia驱动

国内推荐使用GeForce Experience更新驱动

进入WSL键入nvidia-smi

WSL内可正常使用nvidia-smi命令

CUDA

下载安装CUDA11.1

千兆网基本可以跑满

CUDA11.1

同意

accept

此处没有要求安装驱动

Enter

添加环境变量

千万不要指定版本号

Miniconda

~下载安装最新的miniconda

没有出现(base)就source下

创建环境

Pytorch

选择LTS版本

千万注意CUDA版本

直接指定版本

直连速度也不慢

下轮子中

测试一哈

结果如下

大功告成

万一手抖

cuda版本装错了怎么办?

那就再装个对的!

删除旧的

链上新的

查看当前cuda版本

看看对了没

这下对了


WSL2安装Pytorch的评论 (共 条)

分享到微博请遵守国家法律