#创建conda虚拟环境#

命令:

conda create -n cu113_py38 python==3.8

创建成功 如下图:

 

#激活环境#

命令:

conda activate cu113_py38

激活成功 如下图:

#安装cuda113#

命令:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

安装成功 如下图:

 #测试是否安装cuda113成功 #

命令:

import torch 
# 查看torch版本
print(torch.__version__)
# 查看cuda版本
print(torch.version.cuda)
# 查看cuda是否可用
print(torch.cuda.is_available())
# 查看显卡名称
print(torch.cuda.get_device_name())
 

我的测试结果是这样的:

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐