Pytorch安装以及环境配置
首先查看系统的cuda版本的配置 nvcc -V 检查cuda 的编译版本是 nvidia-smi,但是一般使用的是nvcc -V 2、尤其要注意pytorch 、Python 、cuda 版本的对应问题查看Python版本 python --version查看cuda版本:cat /usr/local/cuda/version.txt torch、torchvision、cuda 、python对应版本匹配 3、添加镜像源cond...
原创
2022-04-06 22:07:23 ·
1433 阅读 ·
0 评论