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-2allsobaiduweb~default-1-122083729.142v30pc_rank_34,185v2tag_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()
pytorch-gpu安装_第1张图片
exit() 退出python

nvcc -V 查看cuda版本
nvidia-smi 查看驱动版本
pytorch-gpu安装_第2张图片
5、其他总结
i vim模式下写入
:wq 写完并保存
:q 退出

你可能感兴趣的:(mobaxterm,shell,pytorch,人工智能,python)