首页 > 其他 > 详细

pytorch环境配置

时间:2020-04-29 00:39:25      阅读:95      评论:0      收藏:0      [点我收藏+]

Pytorch安装

Anaconda 安装pytorch

1. anaconda 环境相关命令

查看命令列表: conda info --envs
删除环境:conda remove -n 环境名称 --all
创建环境pytorch_py37:conda create -n pytorch_py37 python=3.7
激活环境:conda activate pytorch_py37
不激活环境:conda deactivate

2. 安装pytorch

  1. 利用 conda create -n pytorch_py37 python=3.7 创建环境
  2. 将其激活conda activate pytorch_py37
  3. 登录到官网:https://pytorch.org/
    技术分享图片
  4. 将命令粘贴到 conda Prompt
    下载慢,
  • 使用清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

另外为了保险起见,建议同时添加第三方conda源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后再去pytorch官网复制下载代码,记得删掉后面的-c pytorch,因为它表示从官网进行下载,不从清华镜像源下。


查看NVIDIA

  1. 搜索并打开
    技术分享图片

  2. 左下角-系统信息:查看驱动版本和CUDA版本
    技术分享图片
    技术分享图片


3.查看安装成功

python

import torch
#gpu是否可用,是则返回true
torch.cuda.is_available() 

技术分享图片

4.pycharm 启用pytorch环境

技术分享图片

pytorch环境配置

原文:https://www.cnblogs.com/kuotian/p/12797931.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!