linux下安装pytorch

一、安装conda，上一篇博客有写

为pytorch创建一个anaconda虚拟环境，环境名字可自己确定。

执行conda create -n pytorch1.7 python=3.11.3，指令中pytorch1.7是我自己的取得名字，

后面3.11.3是我的Ubuntu安装的pyrhon版本，要以实际的来（可以用指令：python -V查看）

二、激活pytorch1.7环境，执行conda activate pytorch1.7(根据2取的名字)

三、安装pytorch的1.7版本执行conda install pytorch torchvision cudatoolkit=10.2 -c pytorch，

10.2为cuda版本号（执行nvcc -V查看），没有就装一个，执行sudo apt install nvidia-cuda-toolkit

四、编辑~/.bashrc 文件，设置使用pytorch1.7环境下的python3.8

执行alias python='/home/alang/anaconda3/envs/pytorch1.7/bin/python3.8'记得改成自己的python路径,然后保存退出后执行： source ~/.bashrc

安装完成！