目录
笔记本型号:机械革命无界16pro
笔记本硬件:cpu——i712700H;gpu——RTX2050(显存4g,cuda数量2048)
软件环境:Win11,WSL2-Ubuntu20.04
更新一下系统,以免版本较老,微软不支持WSL调用gpu,在设置-->windows更新
中即可检查更新
更新显卡驱动,在电脑自带的Nvidia Control Panel-->系统信息
中查看一下自己当前的显卡驱动版本,再去看一下Nvidia官网驱动下载的支持自己gpu型号的显卡驱动, 如果显卡驱动不是最新的, 就下载驱动更新一下
tips: 因为是在物理机上安装驱动所以选win系统的; 下载类型有两种, 有打游戏需求的选GRD, 不打游戏纯生产力的选SD
安装完成后在你的shell配置文件(bash用户就是~/.bashrc, zsh用户就是~/.zshrc)的末尾添加三行命令, 注意根据自己下载cuda版本修改命令中的版本号
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后刷新一下shell配置文件
source ~/.zshrc或~/.bashrc
查看一下cuda版本, 显示正确版本即可
nvcc --version
import torch
# 检查是否可以检测到cuda
torch.cuda.is_available()
# 检查带有cuda的设备数量
torch.cuda.device_count()
# 检查带有cuda的显卡名, 索引从0开始
torch.cuda.get_device_name(0)