标签：安装 Pytorch cuda 版本 PATH 显卡 WSL2 下载

• 在Win11中要干的事儿
• 在WSL中要干的事儿
• 安装Pytorch要干的事儿

笔记本型号：机械革命无界16pro

笔记本硬件：cpu——i712700H；gpu——RTX2050(显存4g，cuda数量2048)

软件环境：Win11，WSL2-Ubuntu20.04

在Win11中要干的事儿

1. 更新一下系统，以免版本较老，微软不支持WSL调用gpu，在设置-->windows更新中即可检查更新

2. 更新显卡驱动，在电脑自带的Nvidia Control Panel-->系统信息中查看一下自己当前的显卡驱动版本,再去看一下Nvidia官网驱动下载的支持自己gpu型号的显卡驱动, 如果显卡驱动不是最新的, 就下载驱动更新一下

   tips: 因为是在物理机上安装驱动所以选win系统的; 下载类型有两种, 有打游戏需求的选GRD, 不打游戏纯生产力的选SD

在WSL中要干的事儿

1. 安装cuda-toolkit, 进入官方页面下载, 注意: 下载之前看一下pytorch目前支持到最高的cuda版本(我安装时还只支持到11.3), 否则你需要重装, 如下即可选择, 会出现命令行操作语句, 使用这两句命令在WSL中安装即可, 中途会让你选需要安装的包, 只选一个cuda-toolkit就行, 另外两个是一些案例没啥用

2. 安装完成后在你的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}}
   

3. 然后刷新一下shell配置文件

   source ~/.zshrc或~/.bashrc
   

4. 查看一下cuda版本, 显示正确版本即可

   nvcc --version
   

安装Pytorch要干的事儿

1. 进入Pytorch官网下载pytorch, 记得要选linux版本, 使用正确版本cuda(建议新建个conda虚拟环境操作), 使用官方给的命令安装即可

2. 验证pytorch能否正确检测cuda

   import torch
   # 检查是否可以检测到cuda
   torch.cuda.is_available()
   # 检查带有cuda的设备数量
   torch.cuda.device_count()
   # 检查带有cuda的显卡名, 索引从0开始
   torch.cuda.get_device_name(0)
   

标签：安装,Pytorch,cuda,版本,PATH,显卡,WSL2,下载
来源： https://www.cnblogs.com/evanzone/p/16397107.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。