1. 写在前面
~~~~~~ 由于作业需要,本人安装了pytorch,在安装的时候遇到了很多的坑,在此处做一些总结。
2. 安装教程
~~~~~~ 安装pytorch特别是GPU版本的pytorch一定要注意版本的匹配。在安装教程上,我找到了一个老哥写是文档,这里就不再造轮子了,直接贴链接:如何安装pytorch
~~~~~~ 我主要阐述一下可能面临的坑。
3. 可能存在的问题
3.1 CUDA安装问题
~~~~~~ cuda在安装时候,会首先解压在一个临时文件夹,然后再安装。安装完成后,其会将临时文件夹删除。因此,第一个安装的位置不是很关键。
~~~~~~ 后面安装的时候,要选择自定义安装,并确保安装位置为自己制定的位置,比如我的是E:\cuda,之后可以按照教程把CuDNN的文件剪贴到指定的文件夹中。
3.2 环境的路径问题
~~~~~~
在安装完成后,我直接自己创建了相关的路径。如果没有创建成功的话,也可以手动创建。
~~~~~~
在创建完成后,尝试命令 nvcc -V,可得到信息如下:显示目前已经安装成功。
3.3 关于包管理的问题
~~~~~~
对于tensorflow以及pytorch的包管理,我还是推荐用Anaconda创建一个虚拟变量,之后在用pychram导入比较方便。关于导入的教程可以参考:
~~~~~~
pychrm导入anaconda环境
4 写在最后
~~~~~~ 谢谢朋友们!