安装pytorch的一些小坑

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 写在最后

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值