1. 写在前面

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

2. 安装教程

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

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

3. 可能存在的问题

3.1 CUDA安装问题

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

安装pytorch的一些小坑

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

安装pytorch的一些小坑

3.2 环境的路径问题

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

3.3 关于包管理的问题

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

4 写在最后

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

相关文章:

  • 2021-07-27
  • 2022-12-23
  • 2022-01-05
  • 2022-12-23
  • 2022-12-23
  • 2022-01-21
  • 2022-12-23
  • 2022-12-23
猜你喜欢
  • 2022-12-23
  • 2021-09-11
  • 2021-06-21
  • 2021-11-22
  • 2021-04-27
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案