windows GeForce MX110 安装驱动、cuda、torch
·
目录
查显卡型号
下载驱动
安装驱动
nvidia-smi查看显卡
低版本torch/cuda
conda安装
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia
执行上面命令的时候，一直卡顿，换成如下pip试下
pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116
我们再换个版本低的torch
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
torch检查cuda/gpu
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
16
0- 0
已为社区贡献1条内容
所有评论(0)