- 博客(8)
- 收藏
- 关注
原创 pytorch安装dgl
pip install dgl-cu118 # 替换为你的 CUDA 版本(如 cu117、cu113 等)print(dgl.backend.cuda.is_available()) # 应输出 True。# 示例:CUDA 11.8 版本。
2025-04-05 22:43:38
90
原创 pytorch中安装apex
无法git clone时,可以直接下载源码的压缩包,然后通过cd进入到文件中,然后pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./如果还报错,尝试python setup.py install --cpp_ext --cuda_ext。注:cuda版本不一致时,注释掉setup.py中的。首先,进入github链接。
2025-04-04 23:16:10
347
原创 linux conda安装tensorflow;以及验证keras
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU'))) #输出GPU可用数量。print(tf.test.is_gpu_available()) #查看tensorflow GPU版本是否可用。print(tf.__version__) #查看tensorflow版本。验证:import tensorflow as tf。
2025-03-19 11:41:55
201
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人