淘先锋技术网

首页 1 2 3 4 5 6 7

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