win10 +Pycharm+Pytorch GPU 环境配置

win10 + Pytorch GPU版本配置

引言

没想到时至今日. 还需要自己写一篇.

PS 建议找连接好一点的网络. 因为要下载的东西比较大. 光Torch 就需要2个G以上. 如果本身没有安装驱动和Cuda的话.需要下载的就更多了.

环境配置

1. 英伟达驱动 传送门()
2. Cuda 传送门()
3. Cudnn 传送门()
4. torch 传送门 建议自己到这个网站上下载. 不要用pip方式.国内网络不友好.

步骤

• 英伟达驱动安装(略)
• Cuda安装(略)
• Cudnn(略)

• 查看需要安装的torch 版本

1. 首先到官网 查看一下所需的版本类型.
   如下方,记住所需的版本号. 以及复制里面的网站https://download.pytorch.org/whl/lts/1.8/torch_lts.html 到浏览器中.
   

2. 到官方下载网(就是前面1.中所说的网站)分别下载对应的版本 torch, torchvision,torchaudio. 其中:
   cu111 代表 cuda 11.1版本
   cp36 代表 python3.6版本
   
   

3. 安装. 用管理员身份打开 powershell, 定位到前面下载的三个文件夹的目录下.
   

4. 用pip 命令依次安装 torch, torchvision torchaudio
   

5. 验证是否安装成功

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出: