在Ubuntu22.4下搭建pytorch深度学习环境

Anacnda安装:这个在其他地方搜索,可以搜到参考oknacUbuntu下安装Anaconda的步骤(带图) - 知乎

安装完毕source ~/.bashrc进入base环境, conda deactivate是推出base环境命令

在Ubuntu22.4下搭建pytorch深度学习环境_第1张图片

 

cuda安装: CUDA Toolkit 11.7 Downloads | NVIDIA Developer

在Ubuntu22.4下搭建pytorch深度学习环境_第2张图片

代码:

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.runsudo sh cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

pytorch安装: PyTorch

 目前没有对应cuda11.7版本,经本人测试,11.6可以用在Ubuntu22.4下搭建pytorch深度学习环境_第3张图片

代码:  pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

 pycharm使用:我是在base环境装的pytorch,新建一个项目,然后配置python解释器

在Ubuntu22.4下搭建pytorch深度学习环境_第4张图片

 测试: 在pycharm中新建一个python文件,import torch没有爆红基本已经配置好了

你可能感兴趣的:(Ubuntu22.4,深度学习,pytorch,python)