CUDA9.0安装pytorch

安装 PyTorch 需要先安装 CUDA，以下是在 Windows 系统下安装 CUDA9.0 和 PyTorch 的步骤：

1. 下载安装 CUDA9.0：访问 https://developer.nvidia.com/cuda-90-download-archive 下载适合自己系统的 CUDA9.0 版本，并按照安装提示进行安装。

2. 安装 cuDNN：访问 https://developer.nvidia.com/cudnn 下载对应版本的 cuDNN，并将其解压到 CUDA9.0 安装路径下的相应目录。例如，如果你在安装 CUDA9.0 时选择的安装路径为 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0，那么将 cuDNN 解压到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 目录下。

3. 安装 Anaconda：访问 https://www.anaconda.com/products/individual 下载适合自己系统的 Anaconda，并按照安装提示进行安装。

4. 创建虚拟环境：打开 Anaconda Prompt，输入以下命令创建一个名为 pytorch 的虚拟环境（也可以自定义名称）：

   conda create --name pytorch python=3.7

5. 激活虚拟环境：输入以下命令激活 pytorch 虚拟环境：

   conda activate pytorch

6. 安装 PyTorch：输入以下命令安装 PyTorch：

   conda install pytorch=0.4.1 cuda90 -c pytorch

7. 验证安装：输入以下命令验证 PyTorch 是否安装成功：

   python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

如果输出版本号和 True，则说明 PyTorch 安装成功并且可以使用 CUDA 进行加速。