【环境搭建】Win10+Anaconda（Python3.7），cudatoolkit10.2+cudnn7.6.5+PyTorch1.5.1

目录

为 PyTorch 创建虚拟环境

更新显卡驱动

添加清华源

安装cudatoolkit和cudnn

使用conda安装PyTorch

测试CUDA是否可用

参考资料

---

这里为https://github.com/yanx27/Pointnet_Pointnet2_pytorch里的项目搭建pytorch版本pointnet++的环境。作者使用的pytorch的版本为1.1.0，但是我发现使用当下（20200703）最新的pytorch1.5.1也可以直接运行作者的代码，所以这里我们使用最新的pytorch1.5.1版本。

首先，安装环境是：操作系统 Win10，已经预先安装了 Anaconda。关于 Anaconda 的安装步骤这里就忽略不讲了，Win10 下安装 Anaconda 非常简单。

为 PyTorch 创建虚拟环境

安装 Anaconda 完毕后，我们在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中，我们通常会根据自己的需求去下载各种相应的框架库，但是可能每个项目使用的框架库并不一样，或使用框架的版本不一样，这样需要我们根据需求不断的更新或卸载相应的库，管理起来相当麻烦。所以通过创建虚拟环境，相当于为不同的项目创建一块独立的空间，在这个空间里，你安装任何库和框架都是独立的，不会影响到外部环境。

总之就是想象成为 PyTorch 单独创建一个空间，这个空间与其它空间（如TensorFlow、其它版本的Python ）互不影响。

创建 PyTorch 虚拟环境步骤：

1.打开 Anaconda 自带的 Anaconda Prompt：

2.打开 Anaconda Prompt 之后，在命令行输入：

conda create --name pytorch1p5 python=3.7

注意，这里的 pytorch1p5 是虚拟环境的名称，可随意取。“pytorch1p5 ”表示我们将安装pytorch1.5版本。

3.然后继续下列命令，进入 pytorch 虚拟环境：

activate pytorch1p5

效果示意图：

<