linux下anoconda 安装pytorch

一、参考教程安装anaconda
二、修改清华源
三、创建pytorch环境并安装 pytorch

一、参考教程安装

anaconda3安装教程(https://blog.csdn.net/tina_ttl/article/details/51762471 )

1、下载anaconda3的安装包:Anaconda3-5.1.0-Linux-x86_64.sh,
下载地址:

https://www.continuum.io/downloads

2、上传至服务器目录下,安装anaconda:

bash /your_dir_where_owns_anaconda/Anaconda2-4.0.0-Linux-x86_64.sh

一直按yes,enter键,一路默认路径安装即可,除了最后一步安装完毕后,会询问你要不要安装vs code,这里no即可,安装完毕。
anocanda安装成功了!

二、修改清华源(可以跳过这步,失败了再回来尝试,主要是加速安装)
百度搜清华镜像

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

修改用户目录下的 .condarc 文件:(没有就新建一个)

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

三、创建pytorch环境并安装
1、重新开一个terminal,使用anaconda创建一个python3.6的环境(名为pytorch):(如果出错“conda: command not found”,就运行 source .bashrc文件,或者重启terminal)

conda create -n pytorch python=3.6

2、激活环境:

source activate pytorch

3、安装pytorch:(这里有多种安装方法)
去pytorch官网(https://pytorch.org/)根据服务器的cuda版本选择对应版本的安装方式
选择linux,pip,cuda版本10.1
这里选择了linux环境下的cuda版本为10.1.(没有GPU则选择None)
按照官网给的安装命令安装:

pip install torch torchvision

4、修改环境中的cuda路径
这里下载的pytorch的cuda版本是10.1,所以要找到自己电脑的cuda10.1的路径,修改用户目录下的.bashrc文件

export PATH="/data/cuda_libs/cuda-10.1/bin:$PATH" 
export CUDA_HOME="/data/cuda_libs/cuda-10.1/"
export LD_LIBRARY_PATH="/data/cuda_libs/cuda-10.1/lib64/"

运行文件,并重启terminal

source .bashrc

5、测试 pytorch是否安装成功:
依次输入:

python 
import torch
print(torch.cuda.is_available())

三、在pytoch环境中使用tensorboard
直接使用pip install tensorboard不行,pip list中虽然有tensorboard这个包,但是使用还是出错。
这种情况可以使用下面语句安装:

python -m pip install tensorboard
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值