PyTorch快速安装与环境配置

1 环境,IDE

  Windows 10,Pycharm
  不借助Anaconda安装PyTorch

2 安装CUDA

  点击到官网下载CUDA：https://developer.nvidia.com/cuda-downloads?
  选择相应的操作系统和版本,例如我的是Windows10 64位，就安装这个版本就行了，其中network 和 local版本的区别就是local版本已经把需要的工具直接下载下来了，而network版本是需要用什么，后期还需要再次下载安装相应工具包，从安装包大小上应该好理解，local版本好几个G,而network版本只有60MB不到

下载好后点击选择安装路径，点击ok进行安装

100%进度后,点击同意并继续:

为了快速安装，就直接选择精简安装了。

若出现重新启动，那么选择马上重新启动，启动后将继续安装。

安装好后点击下一步

点击关闭
检测是否安装成功。在终端输入:nvcc -V即可检测是否安装成功.若是报错 nvcc不是命令不可执行，说明环境变量配置有问题，自行配置环境变量。

3 安装PyTorch

到官网下载PyTorch:https://pytorch.org/get-started/locally/

直接选择pip安装，如果下载慢的话，就用迅雷下载吧,之后再用pip命令安装就行了


再次确认是否安装成功:
在交互界面输入:

import torch
torch.__version__
#是否支持GPU
torch.cuda.is_available()

即可查看安装的PyTorch版本，是否支持GPU。
如下说明安装成功

转载请注明转自:https://leejason.blog.csdn.net/article/details/106651924