【pytorch】conda安装pytorch
Step 1
打开官网:
https://pytorch.org/get-started/locally/
进行选择对应版本:
复制图中命令执行。
Step 2
验证是否安装成功。
执行:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)
输出:
Step 1
打开官网:
https://pytorch.org/get-started/locally/
进行选择对应版本:
复制图中命令执行。
Step 2
验证是否安装成功。
执行:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)
输出: