赞
踩
网上有好多安装cuda的教程,找了好多个,失败了很多天(说多了都是泪),有的比较复杂,作者比较懒,所以想通过纯指令进行安装,于是进行了组合尝试,然后成了,给大家提供一点参考。
感觉自己成功的核心主要是pip指令和conda指令的选择,用pip!!!!
安装位置全在我的虚拟环境pytorch中安装的,下面是我的安装流程:
# 安装CUDA
conda install cudatoolkit=10.2 # 我的电脑显示最高可以11.0,但安全起见,往下安一点
# 安装cudnn,可以自己指定版本,如果不指定版本,在安装CUDA之后,会自动匹配对应版本的cudnn安装
conda install cudnn # 一般不会错,我就没指定
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
如果是10.1的cuda可以参考这个
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
另外出了点小bug,安装完之后import torch进行检查的时候,报错了ImportError: numpy.core.multiarray failed to import,查阅资料发现是pytorch下载自带了一个numpy存在冲突,重新卸载numpy再安装一下就好,没有指定numpy版本
具体操作
#卸载numpy
pip uninstall numpy
# 重新安装numpy
pip install numpy
最后在环境里面输入指令查看安装是否成功
配环境太折磨人了,真的需要慢工出细活,不过感觉配环境也让我更加理解了一些程序运行起来的底层支持吧,虽然应该也很浅薄,进一寸有一寸的欢喜,继续加油。
参考鸣谢:
https://blog.csdn.net/qq_46126258/article/details/112708781 (非常感谢大佬跟我说不能用conda要用pip)
https://blog.csdn.net/Sir666888/article/details/122073675
(简化安装流程)
https://blog.csdn.net/qq_55542305/article/details/126075949
(pytorch清华源安装)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。