Pytorch环境安装_pythorch

必备组件：

• Anaconda3-2021.11-Windows-x86_64
• Python3.9
• cuda_11.1.0_456.43_win10
• cudnn-windows-x86_64-8.3.1.22_cuda11.5-archive
• torch-1.10.0+cu113-cp39-cp39-win_amd64
• torchvision-0.11.1+cu113-cp39-cp39-win_amd64
• 编译器，PyCharm

（torch下载的版本要与本地python的版本对应，不然大概率装不上）

步骤：

1. 安装Anaconda，目录可以自定，要记得在哪，安装时注意勾选上添加到系统环境变量的选项；
2. 安装CUDA，尽量选择默认目录；
3. 安装CUDAnn，是个压缩包，解压后复制其中的文件夹到上一步CUDA的安装目录下粘贴覆盖；
4. 打开PyCharm，新建一个工程，解释器选择对应的python版本，这里使用的是3.9；
5. 工程内新建一个文件，键入如下代码，torch安装之前，import torch应该是标红线报错的；
6. 点击左下角Terminal命令窗口，键入：conda create -n pythorch_gpu python=3.9
   创建gpu运算环境，建立过程中的[y/n]全部键入y；
   
7. 激活虚拟环境，命令：conda activate pytorch_gpu;
8. 激活后，cd到torch与torchversion所在的目录，使用pip install命令进行安装；
9. 安装完成后，在当前工程下，更换解释器为刚刚用conda建立的虚拟环境：
10. 点击右上角的齿轮，点add，添加新建的解释器（刚刚的torch就安装在这里)；
    左侧选择conda，右侧点击指定环境，然后找到conda的安装目录，在anaconda>envs目录下找到python.exe, 点击ok>ok;
11. 最后运行步骤5中的代码，得到如下结果即为安装成功：