当前位置: 代码网 > it编程>前端脚本>Python > Anaconda创建虚拟环境,安装指定版本的pytorch,并配置到pycharm中

Anaconda创建虚拟环境,安装指定版本的pytorch,并配置到pycharm中

2024年08月05日 Python 我要评论
可以选择进行conda安装或者pip安装(conda安装就是命令第一次单词是conda,pip安装同理)可以看见python解释器已经是刚才创建的名为pytorch的环境了。这个就是cuda版本为11.7。没有报错的话就证明安装成功了。

anaconda创建虚拟环境,安装指定版本的pytorch,并配置到pycharm中

创建一个虚拟环境

  1. 打开anaconda prompt命令提示窗口
    image-6
  2. 输入下面命令语句并按下回车创建一个名为pytorch的虚拟环境,并且指定python版本为3.8
    conda create -n pytorch python=3.8
    
  3. 创建好环境之后,输入命令进入刚才创建的环境
    conda activate pytorch
    
    image-7
    可以看见由base环境成功切换到刚才创建的pytorch环境

安装pytorch

  1. 进入pytorch官网

  2. 点击首页getsart,然后选择previous pytorch versions
    image-8
    image-9

  3. 键盘按住ctrl+f搜索想要安装的torch版本,比如我想安装torch版本为1.13.1
    image-10

    可以选择进行conda安装或者pip安装(conda安装就是命令第一次单词是conda,pip安装同理)

  4. 将命令复制到anaconda prompt中,按下enter键等待安装完成即可,这个命令同时也会安装与torch版本相对应的torchvision,torchaudio版本

  5. 有英伟达显卡的可以选择安装有cuda版本的,也可以选择cpu版本,cpu版本相比于cuda版本占用内存要小很多,没有英伟达显卡的就只能选择cpu版本
    image-11
    这个就是cuda版本为11.7

测试是否安装成功

  1. 在anaconda prompt中输入
python
  1. 导入torch包
import torch

image-12

没有报错的话就证明安装成功了
3. 退出则需要输入

exit()

将配置好的环境作为解释器配置到pycharm中

  1. 打开pycharm,依次点击file → setting → project → python interpreter,点击右上角齿轮,选择add interpreter
    image-13
    image-14

  2. 选择existing environment,点击右边的三个点,选择新创建环境下的python.exe文件(一般在anaconda软件安装目录下 → envs → pytorch文件夹下,这是我的路径,供参考 “d:\anaconda\envs\pytorch”)
    image-15
    image-16

  3. 选择完毕后点击确定确定就可以了
    image-18
    可以看见python解释器已经是刚才创建的名为pytorch的环境了

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com