ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

基于anaconda3的Pytorch环境搭建

2022-09-12 12:02:31  阅读:299  来源: 互联网

标签:anaconda3 python Pytorch pytorch conda cuda 版本 官网 搭建


方法一

  1. 安装anaconda3,版本选择新的就行
  2. 打开anaconda prompt创建虚拟环境conda create -n pytorch_gpu python=3.9,pytorch_gpu是环境名称,可自行选取,python=3.9是选择的python版本,可自行选择,conda会自动下载选择的python版本
  3. 接下来去pytorch官网 https://pytorch.org/get-started/locally/
  • 直接选择最新的cuda安装,把conda命令复制到环境下执行即可
  • 安装结束执行torch.cuda.available()返回true则成功

(这个方法最简单但可能会不行,我自己环境下是没问题的,尽管的我的显卡驱动版本不支持cuda的高版本,但是也可以,不行换第二个方法)

方法二

  1. 安装anaconda3,版本选择新的就行
  2. 打开anaconda prompt创建虚拟环境conda create -n pytorch_gpu python=3.9,pytorch_gpu是环境名称,可自行选取,python=3.9是选择的python版本,可自行选择,conda会自动下载选择的python版本
  3. 打开cmd按照下图输入查看显卡驱动版本
    image
  4. 查看显卡驱动版本与cuda兼容版本
    image
    (链接:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
  5. 选择合适的cuda版本,比如我这里驱动版本为457,那么可以选择任何比457小的cuda版本。(我这里>=457的都用不了)
  6. 接下来去pytorch官网(https://pytorch.org/get-started/previous-versions
  • 先进入自己创建的虚拟环境
  • 在pytorch官网中找到匹配自己cuda版本的conda命令复制到自己的conda环境下安装(这一步没必要切换源,我从官网直接下每次都成功了)
  • 安装结束执行torch.cuda.available()返回true则成功

补充

  • 每必要按照大部分网上的教程安装cuda再安装cudann,因为pytorch官网的conda命令包含了cudatoolkit

标签:anaconda3,python,Pytorch,pytorch,conda,cuda,版本,官网,搭建
来源: https://www.cnblogs.com/sxq-blog/p/16685850.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有