我的电脑GPU是AMD Radeon(TM) Graphics,无法直接使用CUDA,下面给出一个可以替代的方案。

1.进入Anaconda prompt,输入conda env list,大部分应该只有一个base环境。

(base) D:\AItools>conda env list

# conda environments:
#
base                 * D:\AItools\Anaconda3

2.创建虚拟环境deepbase

(base) D:\AItools>conda create -n deepbase python=3.10 -y

!!注意!!:本文是下载torch-directml去替代torch,而torch-directml 要求 Python的版本在 3.7-3.10,如果你的python版本过高,可能会在后面下载时出错。因此这里面python版本建议使用3.10。

创建成功后:

3.启用deepbase环境

(base) D:\AItools>conda activate deepbase

启用成功的标志是(base)变成了(deepbase)

4.进去deepbase后查看python版本

(deepbase) D:\AItools>python --version
Python 3.10.16

5.查看是否有torch-directml

(deepbase) D:\AItools>pip show torch-directml
WARNING: Package(s) not found: torch-directml

6.安装torch-directml,-i 后面是清华镜像

(deepbase) D:\AItools>pip install torch-directml -i https://pypi.tuna.tsinghua.edu.cn/simple

7.检验是否安装完成

(deepbase) D:\AItools>python
Python 3.10.16 | packaged by conda-forge | (main, Dec  5 2024, 14:07:43) [MSC v.1942 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch_directml
>>> torch_directml.is_available()
True
>>>

查看GPU

>>> torch_directml.device_name(0)
'AMD Radeon(TM) Graphics\x00'

到此说明torch-directml完成安装。

Logo

火山引擎开发者社区是火山引擎打造的AI技术生态平台,聚焦Agent与大模型开发,提供豆包系列模型(图像/视频/视觉)、智能分析与会话工具,并配套评测集、动手实验室及行业案例库。社区通过技术沙龙、挑战赛等活动促进开发者成长,新用户可领50万Tokens权益,助力构建智能应用。

更多推荐