1 安装显卡驱动

2 查看安装  nvidia-smi  是一个用于NVIDIA GPU(图形处理单元)管理和监控的命令行实用程序

3 安装Anaconda 

4 安装 Cuda与CuDNN

5 安装Python  

6  安装pytorch GPU版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

7 测试效果

8

10

11 运行手写数字识别成功

12 训练OK

在pycham运行OK,但是在ubuntu出错,需要设置下路径

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐