当前位置:   article > 正文

anconda虚拟环境安装cuda10.2 懒人篇_conda 安装cuda

conda 安装cuda

网上有好多安装cuda的教程,找了好多个,失败了很多天(说多了都是泪),有的比较复杂,作者比较懒,所以想通过纯指令进行安装,于是进行了组合尝试,然后成了,给大家提供一点参考。

感觉自己成功的核心主要是pip指令和conda指令的选择,用pip!!!!

可以先参考这个博主的方法查询自己能安装的cuda版本

安装位置全在我的虚拟环境pytorch中安装的,下面是我的安装流程:

  1. 安装cuda
# 安装CUDA
conda install cudatoolkit=10.2 # 我的电脑显示最高可以11.0,但安全起见,往下安一点
  • 1
  • 2
  1. 安装cudnn
# 安装cudnn,可以自己指定版本,如果不指定版本,在安装CUDA之后,会自动匹配对应版本的cudnn安装
conda install cudnn # 一般不会错,我就没指定
  • 1
  • 2
  1. 安装pytorch
    采用了清华源,安装对应10.2cuda的pytorch
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
  • 1

如果是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
  • 1

另外出了点小bug,安装完之后import torch进行检查的时候,报错了ImportError: numpy.core.multiarray failed to import,查阅资料发现是pytorch下载自带了一个numpy存在冲突,重新卸载numpy再安装一下就好,没有指定numpy版本
具体操作

#卸载numpy
pip uninstall numpy

# 重新安装numpy
pip install numpy
  • 1
  • 2
  • 3
  • 4
  • 5

最后在环境里面输入指令查看安装是否成功
终于是true了TAT
配环境太折磨人了,真的需要慢工出细活,不过感觉配环境也让我更加理解了一些程序运行起来的底层支持吧,虽然应该也很浅薄,进一寸有一寸的欢喜,继续加油。

参考鸣谢:
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清华源安装)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/398005
推荐阅读
相关标签
  

闽ICP备14008679号