#创建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())
我的测试结果是这样的: