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

最新推荐文章于 2024-03-18 12:08:24 发布

雨_刃

最新推荐文章于 2024-03-18 12:08:24 发布

阅读量1.5k

点赞数

分类专栏：软件安装文章标签： pytorch linux

本文链接：https://blog.csdn.net/YENTERTAINR/article/details/121054775

版权

本文档介绍了如何在Linux系统中配置conda环境，包括修改.bashrc文件以设置CUDA路径，更换conda源为清华大学镜像，并在环境中安装PyTorch。此外，还讲述了激活虚拟环境的方法以及在遇到问题时的解决步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

安装命令：
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/an