Linux 安装 pytorch(cuda=10.1),自己记录个人安装关键步骤,不做学习参考。


安装命令:
conda install pytorch 1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch -y

1.修改/root/.bashrc中的环境配置:

.bashrc文件只做了最后export的添加。

# .bashrc

# User specific aliases and functions
alias rm='rm -i'
alias cp='cp -i'
alias mv='mv -i'
alias vi='vim'
alias wc='watch -n 0.1 nvidia-smi'
alias ppy='ps -ef | grep python'


if [ -f /etc/bashrc ]; then
	. /etc/bashrc
fi

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/root/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/root/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/root/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/root/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

export TERMINFO=/usr/share/terminfo

# 最后这句话是添加的
Export PATH="/root/anaconda3/bin:$PATH"

2.换源:

进入/root/.condarc文件(这个是索引文件)中,修改成下面的源:

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - 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

3.上述完成后,

3.1 进入自己的虚拟环境conda acitvate myTestEnv
3.2 myTestEnv]# python
3.2 >>> import torch
3.3 >>> torch.cuda.is_available()

4.如果存在上述配置结束后,还有问题:

4.1比如,说li****.iso找不到路径,这说明安装上了,但是环境配置出问题。重复第一步操作。
4.2 重复之后,如果需要刷新,可以source .bash 操作,即,更新资源。

5 Windows和Linux的虚拟环境切换区别:

5.1 Linux的虚拟环境切换:conda activate Your_EnvName
ps:这个Your_EnvName不要加路径。比如/home/your_EnvName是不对的。使用root的默认的即可

5.2 Windows的虚拟环境切换:
需要加上路径:
比如:activate D:\AworkStation\Virtual\Pytorch1102
这个需要添加路径

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值