码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch环境配置

时间:2020-04-28 23:11:57      阅读:295      评论:0      收藏:0      [点我收藏+]

标签:tun   http   添加   imp   打开   成功   当前目录   container   ffline   

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环境配置

标签:tun   http   添加   imp   打开   成功   当前目录   container   ffline   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!