(Ubuntu Linux)Conda安装Pytorch2.0-Python3.8-Cuda11.7
一、安装Anaconda
安装包下载:
https://repo.anaconda.com/archive/
1.选择合适的安装包,每个版本中对应不同的python版本,根据需要进行选择。例如我选择的是(Anaconda3-2020.11-Linux-x86_64.sh对应 python3.8.5)
2.将下载的Anaconda3-xxx.sh文件放置在某个路径下并cd进入该目录
执行sudo bash Anaconda3-XXX.sh
# bash 对安装包进行安装
安装过程中会要求输入指令,全选yes即可。
安装完成后需要执行一次source ~/.bashrc
最后通过conda --version检查安装是否完成。
二、安装Cuda
安装包下载:
https://developer.nvidia.com/cuda-toolkit-archive
1.选择合适的安装包,根据pytorch的需求进行选择。例如我选择的是(CUDA Toolkit 11.7.0),该版本对应pytorch2.0
随后选择安装包下载并安装,官方在网页中给出了下载和安装的方式,也可将链接复制到浏览器中下载好后手动安装。
三、安装Pytorch
python,cuda版本均安装正确后才能顺利安装Pytorch。
进入Pytorch官网 https://pytorch.org/ 并获取安装命令,根据需求进行选择。
如果安装失败,可跟换清华源进行安装。