安装pytorch的一些小坑

安装pytorch的一些小坑

  • 1. 写在前面
  • 2. 安装教程
  • 3. 可能存在的问题
    • 3.1 CUDA安装问题
    • 3.2 环境的路径问题
    • 3.3 关于包管理的问题
  • 4 写在最后

1. 写在前面

       ~~~~~~       由于作业需要,本人安装了pytorch,在安装的时候遇到了很多的坑,在此处做一些总结。

2. 安装教程

       ~~~~~~       安装pytorch特别是GPU版本的pytorch一定要注意版本的匹配。在安装教程上,我找到了一个老哥写是文档,这里就不再造轮子了,直接贴链接:如何安装pytorch

       ~~~~~~       我主要阐述一下可能面临的坑。

3. 可能存在的问题

3.1 CUDA安装问题

       ~~~~~~       cuda在安装时候,会首先解压在一个临时文件夹,然后再安装。安装完成后,其会将临时文件夹删除。因此,第一个安装的位置不是很关键。

安装pytorch的一些小坑_第1张图片

       ~~~~~~       后面安装的时候,要选择自定义安装,并确保安装位置为自己制定的位置,比如我的是E:\cuda,之后可以按照教程把CuDNN的文件剪贴到指定的文件夹中。

安装pytorch的一些小坑_第2张图片

3.2 环境的路径问题

       ~~~~~~        在安装完成后,我直接自己创建了相关的路径。如果没有创建成功的话,也可以手动创建。安装pytorch的一些小坑_第3张图片
       ~~~~~~        在创建完成后,尝试命令 nvcc -V,可得到信息如下:显示目前已经安装成功。
安装pytorch的一些小坑_第4张图片

3.3 关于包管理的问题

       ~~~~~~        对于tensorflow以及pytorch的包管理,我还是推荐用Anaconda创建一个虚拟变量,之后在用pychram导入比较方便。关于导入的教程可以参考:
       ~~~~~~        pychrm导入anaconda环境

4 写在最后

       ~~~~~~       谢谢朋友们!

你可能感兴趣的:(Python学习,python,深度学习,pytorch)