pytorch入门到项目（—）

pytorch入门到项目（—）

一、pytorch的环境

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

python	3.7
anaconda	5.3.1
cuda	10.0
pycharm community	PyCharm 2020.1 (Community Edition)

二、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