一、检查版本是否匹配
使用Python进入控制台,然后执行:
>>>print(torch.cuda.is_available())
如果输出为 False,则说明不匹配
退出Python环境,再检查系统版本:
exit()
nvcc -V
二、安装
去官网 PyTorch 获取pip安装命令,按照可视化进行选择,复制下方的一键安装命令进行安装
选择适合自己的 平台 和版本,在控制台执行安装即可,注意在windows下使用的是pip不是pip3
一、检查版本是否匹配
使用Python进入控制台,然后执行:
>>>print(torch.cuda.is_available())
如果输出为 False,则说明不匹配
退出Python环境,再检查系统版本:
exit()
nvcc -V
二、安装
去官网 PyTorch 获取pip安装命令,按照可视化进行选择,复制下方的一键安装命令进行安装
选择适合自己的 平台 和版本,在控制台执行安装即可,注意在windows下使用的是pip不是pip3