【小白学习记录2】Centos7下conda安装CUDA版本的pytorch

本篇参考：https://blog.csdn.net/qq_20622615/article/details/80667328

1.配置虚拟环境

（OpenCV不支持Anaconda默认的python3.8，因此安装指定3.6版本python的pytorch）

conda create -n pytorch python=3.6

2.激活虚拟环境

conda activate pytorch

3.安装pytorch

进入pytorch官网选择版本 https://pytorch.org/

我的CUDA版本为11.2，选择了11.1conda安装

4.修改bashrc文件

vim ~/ .bashrc 
export LD_LIBRARY_PATH=/export/home/user/anaconda3/envs/pytorch/lib:$LD_LIBRARY_PATH #添加到最后 编辑完以后ESC ：wq保存并退出

应用所更改的变量

source .bashrc

5.验证是否成功

conda activate pytorch激活环境后，输入python，查看到此时python版本是3.6. 输入

import torch

没有报错

再输入print(torch.cuda.is_available())

显示True，安装成功

#移除虚拟环境 conda remove -n pytorch --all