Windows 10 下 PyTorch安装及调试

首先要注意不要直接使用pip install torch,会报错。

打开PyTorch官网:https://pytorch.org/get-started/locally/

根据自己的需要对操作系统、安装版本及安装工具进行选择,网页会自动生成安装命令,打开命令行输入安装命令。

安装完成后可使用如下代码进行调试

import torch

print(torch.__version__)        #torch版本
print('gpu:', torch.cuda.is_available())   #gpu加速是否可用

pip工具的安装使用可参考博客:https://blog.csdn.net/sinat_40471574/article/details/91354263 

发布了34 篇原创文章 · 获赞 26 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/sinat_40471574/article/details/104416453