查看电脑显卡支持的CUDA版本
- 开始搜索控制面板
- 从控制面板打开NVIDIA控制面板
- 打开左下角‘系统信息’,‘组件’,查看可知本机显卡支持最高CUDA版本为11.0.208

利用Anaconda创建虚拟环境
过程参考我另一篇博文,配置好numpy等其他依赖包。
激活虚拟环境
activate EnvName
查询pytorch和对应CUDA
进入Pytorch官网,根据需要的Pytorch和CUDA版本查看conda命令。
此处我配置Pytorch1.10+CUDA10.2,命令为:
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
运行太慢或网络连接失败,手动从以下4个网址查询需要的包,如果有则指定对应channel
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkg

本文档介绍了如何检查电脑显卡支持的CUDA版本,利用Anaconda创建虚拟环境,并在环境中安装PyTorch 1.10.0+CUDA10.2以及CUDA版本的DGL。首先,通过NVIDIA控制面板查看CUDA版本为11.0.208。接着,详细说明了如何使用conda命令在虚拟环境中安装PyTorch、torchvision和torchaudio。当从官方渠道安装遇到问题时,可尝试从清华大学开源软件镜像站下载所需包。最后,给出了安装CUDA10.2版本DGL的步骤,包括直接命令和手动查找镜像源。

52

被折叠的 条评论
为什么被折叠?



