在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