pytorch和tensorflow用很多人要问,如果你是工程师更推荐使用TensorFlow2.0 如果你想从事科研学术或者学生可以选择pytorch可以满足快速迭代和发表文章的需求。
安装配置环境是劝退的第一步,按着下面步骤一定可以!
现在来说在win10系统下 pytorch的下载安装:
1. 安装anaconda3(很多人推荐下载安装但是我在anaconda中下载pytorch失败了,下面步骤成功)
2. 安装CUDA:
需要注册,下载时间长
2.2环境变量配置
系统属性环境变量里新增5个变量
CUDA_BIN_PATH %CUDA_PATH%\bin
CUDA_LIB_PATH %CUDA_PATH%\lib\Win64
CUDA_SDK_BIN %CUDA_SDK_PATH%\bin\Win64
CUDA_SDK_LIB %CUDA_SDK_PATH%\common\lib\Win64
CUDA_SDK_PATH C:\cuda\cudasdk\common
2.3测试是否配置成功
Cmd中输入:nvcc -V 出如下图配置成功
3、安装pytorch
3.1安装
复制run this command中的内容在cmd中下载
安装成功CUDA之后再安装pytorch非常快
3.2最后检测下是不是安装成功
在python环境中 cmd输入python
之后输入import torch没有报错就没有什么问题
接下来,输入 torch.cuda.is_available,如果没有问题,意味可以使用 GPU