-
安装pytorch环境：windows10+GPU(cuda+cudnn)+pytorch(torch+torchvision)+pycharm
- cuda下载网站：developer.nvidia.com/cuda-downlo…
- cudnn下载网站：developer.nvidia.com/cudnn-downl…
- 多版本cuda切换：blog.csdn.net/sinat_38132…
- torch+torchvision下载网站：download.pytorch.org/whl/torch_s…，版本为：torch-1.9.0+cu102-cp39-cp39-win_amd64.whl和torchvision-0.10.0+cu102-cp39-cp39-win_amd64.whl
-
注意：ERROR: torch-1.9.0+cu102-cp39-cp39-win_amd64.whl is not a supported wheel on this platform.
- 发生这个提醒是代表版本不对，python3.6应该对应cp36版本的
-
软件安装目录：anaconda3直接安装在D盘，cude直接安装在D盘，虚拟环境在E:\anaconda3\envs，torch和torchvision需要在anaconda prompt安装
- 第一步：torch和torchvision安装步骤：conda create -n pytorch(虚拟环境名称可自行更改) python=3.6(安装的python版本可自行更改)
- 第二步：activate pytorch命令激活虚拟环境
- 第三步：切换到torch和torchvision的目录执行命令安装：pip install "torch-1.9.0+cu102-cp39-cp39-win_amd64.whl"和pip install "torchvision-0.10.0+cu102-cp39-cp39-win_amd64.whl"
-
在pycharm引入虚拟环境运行py脚本：新建一个PyCharm工程，然后File -> Settings -> Project Interpreter, 这时会发现项目解释器是PyCharm自带的或者是Anaconda自带的python.exe。然而，我们需要使用pytorch的库，所以点击右边代表设置符号，点击add, 选中Existing environment，找到目录并选中：E:\anaconda3\envs\pytorch\python.exe 作为该项目的解释器。
-
验证
- import torch
- print("hello pytorch{}".format(torch.version)//验证torch
- print(torch.cuda.is_available())//验证GPU
