Anacnda安装:这个在其他地方搜索,可以搜到参考oknacUbuntu下安装Anaconda的步骤(带图) - 知乎
安装完毕source ~/.bashrc进入base环境, conda deactivate是推出base环境命令
cuda安装: CUDA Toolkit 11.7 Downloads | NVIDIA Developer
代码:
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可以用
代码: pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
pycharm使用:我是在base环境装的pytorch,新建一个项目,然后配置python解释器
测试: 在pycharm中新建一个python文件,import torch没有爆红基本已经配置好了
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)