Intel A770 GPU深度学习环境搭建（Linux）
Intel A770 GPU在Linux系统下的深度学习环境搭建指南
views
| comments
目录#
step 1 安装GPU驱动#
前往Intel GPU驱动网站 ↗，安装驱动。
step 2 安装PyTorch#
我选择的是使用pip进行安装。你可以先使用conda创建一个虚拟环境。
安装命令
release Version
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/xpunightly Version
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu完成后进行测试。
python -c "import torch; print(torch.__version__); [print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"最后成功识别到显卡即为安装成功。
前往example ↗进行愉快的玩耍吧。
step 3 安装intel-gpu-tools。#
安装intel-gpu-tools可以帮助你更好地监控和管理Intel GPU。
sudo apt install intel-gpu-tools安装完成后，你可以使用以下命令查看GPU状态：
sudo intel_gpu_top来查看GPU的使用情况。