pytorch的环境配置

pytorch的环境配置

参考：https://blog.csdn.net/weixin_44791964/article/details/106037141

我的pytorch环境配置需求：

python 3.6.5
pytorch 1.9.0 
cuda 10.2
numpy 1.19.2
matplotlib 3.1.0

1.Anaconda的下载和安装 （参考上面链接） 2.Cudnn和CUDA的下载和安装（参考上面链接）

CUDA（百度网盘有cuda10.2版本，可以直接下载使用。）

3.配置torch环境

第一步：

conda create –n pytorch python=3.6.5

第二步：

activate pytorch

第三步：下载gpu版本的pytorch

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

第四步：再安装其他需要的包，比如numpy等

最后

在pycharm中运行测试是否成功

import torch
print(torch.__version__)  //pytorch是cpu还是gpu版本
print( torch.zeros(1).cuda())
print(torch.cuda.is_available())  //电脑的 GPU 能否被 PyTorch 调用