WSL2调用Windows中的GPU版Pytorch

在nvidia官网可以下载WSL2版本cuda,但最低cuda版本为11.3,而我的电脑最高支持的10.2版本的cuda驱动,因此即使安装了cuda也无法安装相应的驱动。

网上有很多使用镜像的教程,但都失败了,不想尝试了,都不知道怎么清理内存了,cuda文件都删除卸载了,还是少了10G内存。

wsl2命令行输入 echo $PATH 命令是可以看到很多windows系统路径,因此我们完全可以使用windows下gpu版本的pytoch。配置如下:

在命令行输入 code .即可安装vscode,其实就是调用windows下的vscode;

1. 创建python测试文件

import torch
print(torch.__version__)
print(torch.randn((1, 2, 3)))
print(torch.cuda.is_available())

2. 配置settings.json文件,如最后一行所示配置对应的python路径,就是选择windows上GPU版本的pytorch:

"code-runner.executorMap": {
        "javascript": "node",
        "java": "cd $dir && javac $fileName && java $fileNameWithoutExt",
        "c": "cd $dir && gcc $fileName -o $fileNameWithoutExt
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值