在window的pycharm里的pytorch框架上使用AMDgpu加速深度学习训练

AMD加速训练

介绍

AMD加速深度学习使用dml框架pytorch

软件架构

软件架构说明 python3.8版本可用 在pycharm里

使用说明

  1. 使用torchvision0.9.0
  2. 使用pip install torchvision==0.9.0时会自动下载torch1.8.0版本
  3. 需要自行pip uninstall torch

安装教程

首先创建环境python3.8版本 推荐下载anaconda管理环境 本步骤基于anaconda

第一种方法

打开pycharm点击file,点击setting  点开interpreter,在点add  点击conda环境,新建环境,不要点make availale to all project  查看环境是否新建成功,打开anaconda的文件夹,打开envs,看是否有刚刚新建的环境名  这样环境就新建成功了

在window的pycharm里的pytorch框架上使用AMDgpu加速深度学习训练_第1张图片

 在window的pycharm里的pytorch框架上使用AMDgpu加速深度学习训练_第2张图片

在window的pycharm里的pytorch框架上使用AMDgpu加速深度学习训练_第3张图片在window的pycharm里的pytorch框架上使用AMDgpu加速深度学习训练_第4张图片

第二种方法

直接在anaconda里新建  

你可能感兴趣的:(python,pytorch,深度学习)