pytorch入门到项目（—）

一、pytorch的环境

本身项目采用win10系统+pycharm+anaconda+cuda。
其中版本为

二、pytorch的安装

在pytorch.org的官网上选择相应的版本环境。
在conda命令行中输入输入代码
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch?
安装完成后测试是否安装成功

import torch

print(torch.__version__)
print("gpu:",torch.cuda.is_available())

显示如下输出时表示安装成功

1.4.0
gpu: True
Process finished with exit code 0

pytorch入门到项目（—）

原文：https://www.cnblogs.com/wuxero/p/13201247.html