LINUX服务器搭建pytorch环境

LINUX服务器搭建pytorch环境

1. 安装Anaconda

1.1 下载anaconda的安装包

这里我们需要在官网上查找自己需要的版本,地址链接在下面:

https://repo.anaconda.com/archive/

这是我选择的版本,然后我们在控制台输入这句话:

wget https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh

1.2 安装anaconda

安装包下载后,用bash 文件名命令安装

bash Anaconda3-5.2.0-Linux-x86_64.sh

然后出现下面图所示,点击回车
在这里插入图片描述
此时显示Anaconda的信息,并且会出现More,继续按Enter,直到如下图所示:
在这里插入图片描述
输入yes
在这里插入图片描述
继续回车
在这里插入图片描述
这里选择no,跳过,以免污染环境。
安装结束后操作:
那你需要自己到这个文件夹设置你安装Anaconda路径

vi /home/服务器用户名/.bashrc

在最后一行添加:

export PATH=/home/服务器用户名/anaconda3/bin:$PATH

ps.先按 i 进入insert模式,然后添加,之后按esc,然后输入:wq为强行保存退出

完成安装以及检测是否安装成功

打开新的终端后,进入自己的文件夹目录下,输入anaconda -V(注意a要小写,V要大写),conda -V ,显示版本信息,若显示则表示安装成功。

conda -V

2.Anaconda安装Pytorch

2.1 创建虚拟环境

conda create -n pytorch python=3.7
(pytorch 为环境名称)

2.2 激活环境

conda activate pytorch
(pytorch 为所创建的环境名称)

2.3 检测环境是否安装好

conda info --envs

Linux环境中搭建CUDA的步骤如下: 1. 首先,需要安装CUDA和CUDNN。根据需要安装的CUDA版本,参考适当的安装教程以及官方文档。可以从NVIDIA官方网站下载CUDA安装包,并按照说明进行安装。 2. 安装完毕后,可以使用命令nvcc -V和nvidia-smi来验证CUDA的版本。在某些情况下,通过这些命令显示的CUDA版本可能与实际运行时使用的版本不同。可以使用import torch和torch.utils.cpp_extension.CUDA_HOME命令来查看PyTorch实际使用的CUDA版本。 3. 如果想切换CUDA版本,可以在.bashrc文件中进行修改。打开.bashrc文件,添加以下代码: ``` export PATH="/usr/local/cuda-<version>/bin/:$PATH" export LD_LIBRARY_PATH="/usr/local/cuda-<version>/lib64:$LD_LIBRARY_PATH" export CUDA_HOME=/usr/local/cuda-<version> ``` 其中,<version>替换为你想要使用的CUDA版本号。保存并关闭.bashrc文件。 4. 为了使修改后的.bashrc文件生效,需要在终端中运行以下命令: ``` source ~/.bashrc ``` 这样,修改后的CUDA环境就会生效。 5. 最后,可以使用nvcc -V和nvidia-smi命令来验证CUDA版本是否已成功切换。 需要注意的是,上述步骤仅适用于在用户自己的环境中切换CUDA版本。如果在共享服务器上使用,可能需要在/etc/profile文件中进行修改,以改变默认的CUDA版本。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [【转载】Linux安装CUDA 和 cuDNN,以及CUDA个人环境和系统环境的切换](https://blog.csdn.net/a_piece_of_ppx/article/details/129477191)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [服务器Linux环境配置cuda(非管理员)](https://blog.csdn.net/ffftttxxx/article/details/125910239)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值