[Python知识库]pytorch-gpu安装

1、在centos里安装anaconda3：
https://blog.csdn.net/Mr_Blackk/article/details/120621190

2、
新建环境： conda create -n py38 python=3.8

进入环境：conda activate py38（py38是自己建的环境名）

退出环境：conda deactivate

3、
先在pytorch官网下载pytorch（若anaconda替换为清华源）。即便下gpu版的，下载完用conda list一看，下下来的还是CPU版本的。解决方案，手动替换为GPU版本的。见如下博客：
https://blog.csdn.net/weixin_43868021/article/details/122083729?ops_request_misc=&request_id=&biz_id=102&utm_term=centos%E4%B8%8Bconda%E5%AE%89%E8%A3%85pytorch%E6%98%AFCPU%E7%89%88%E6%9C%AC%E5%AF%BC%E8%87%B4c&utm_medium=distribute.pc_search_result.none-task-blog-2_allsobaiduweb~default-1-122083729.142^v30pc_rank_34,185^v2tag_show&spm=1018.2226.3001.4187

总结：
mobaxterm离线下载安装软件
1）从清华源镜像下载自己需要的软件包
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
2）上传到mobaxterm的一个文件夹里
3）cd到这个文件夹
conda install --offline 文件名

4、查看是否成功
import torch
torch.version
torch.cuda.is_available()

exit() 退出python

nvcc -V 查看cuda版本
nvidia-smi 查看驱动版本

5、其他总结
i vim模式下写入
:wq 写完并保存
:q 退出